PyTorch 图像处理及格式转换

图像格式转换

1 、tensor to PIL image

from torchvision import transforms 
count = 0
def TensortoImg(img):
      global count
      img = img.cpu()
      img = transforms.ToPILImage()(img[0])
      # img.show()
      img.save('count.jpg')
      count += 1
发布了16 篇原创文章 · 获赞 7 · 访问量 1万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览