仅作为记录,大佬请跳过。
直接上代码
torch.save(pred_mask[i].cpu().detach(), '/output/pred_mask.pth')
torch.save(torch.unsqueeze(mask[i], 0).cpu().detach(), '/output/mask.pth')
展示
***********************
加载的代码
q=torch.load('/output/pred_mask.pth')
qq=torch.load('/output/mask.pth')