pytorch~改变tensor的值(对mask进行0\1化操作)
1 2 3 4 5 6 7 | new_mask_image = torch.zeros([inst_map.shape[ 0 ],inst_map.shape[ 1 ],inst_map.shape[ 2 ],inst_map.shape[ 3 ]], dtype = torch.float32,device = inst_map.device) for i in range (inst_map.shape[ 0 ]): f_mask = inst_map [i,:,:,:] zero = torch.zeros_like(f_mask) one = torch.ones_like(f_mask) f_mask = torch.where(f_mask> - 1.0 , one, zero) new_mask_image[i,:,:,:] = f_mask |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步