【原】PIL图片经过缩放后再tensor然后传入add_image

表白：黑白圣堂血天使，天剑鬼刀阿修罗。 
讲解对象：/PIL图片经过缩放后再tensor然后传入add_image
作者：融水公子 rsgz
===

PyTorch技巧 

PIL图片经过缩放后再tensor然后传入add_image

from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter

pic_one = r"D:\2-code\1-AI_pro\Mangguo_rpo\8-selenium\9-PyTorch\shujuji\hymenoptera_data\train\ants\5650366_e22b7e1065.jpg"
img_pil = Image.open(pic_one)
print(img_pil.size)

writer = SummaryWriter(r"D:\2-code\1-AI_pro\Mangguo_rpo\8-selenium\9-PyTorch\logs")
trans_resize = transforms.Resize((512,512))
img_resize = trans_resize(img_pil)
print(img_resize)

trans_totensor = transforms.ToTensor()
img_tensor=trans_totensor(img_resize)

writer.add_image("img_tensor", img_tensor)
writer.close()

代码结果

(500, 375)
<PIL.Image.Image image mode=RGB size=512x512 at 0x2454B9E8C10>

tensorboard

=== 

公众号：不浪仙人

谢谢大家的支持！可以点击我的头像，进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录，里面真的有很多优秀的文章，欢迎大家的到来。
---

';