os.mkdir('/home/ubuntu/DEKR-main1/11/'+str(i))标签:11,tensor,main1,image,保存,str,ubuntu,home,图片 From: https://www.cnblogs.com/hahaah/p/17145694.html
torchvision.utils.save_image(image,'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(i)+'.png')
for k in range(19):
torchvision.utils.save_image(heatmap[0,k,:,:],'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(k)+'h'+'.png')
for k in range(38):
torchvision.utils.save_image(offset[0,k,:,:],'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(k)+'off'+'.png')