栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

transforms.ToTensor()_transform normalize?

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

transforms.ToTensor()_transform normalize?

class ToTensor:
    """Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor. This transform does not             
    support torchscript.

对于ToTensor的使用测试, 

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

writer = SummaryWriter("logs")
img = Image.open("C:/Users/Mxx/PycharmProjects/pythonProject4/images_set/GOGOGOGO.png")
print(img)

trans_totensor = transforms.ToTensor() #首先创建一个对象,来使用ToTensor的方法
img_tensor = trans_totensor(img)
writer.add_image("ToTensor",img_tensor)
writer.close()

writer.add_image需要接收一个img_tensor的数据类型

    def add_image(self, tag, img_tensor, global_step=None, walltime=None,     
               dataformats='CHW'):
        """Add image data to summary.

使用tensorboard来查看,转成没。 

 tensorboard --logdir=temp/logs


TensorFlow installation not found - running with reduced feature set.
Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all
TensorBoard 2.8.0 at http://localhost:6006/ (Press CTRL+C to quit)

成功插入这个img的tensor类型。 

对其进行标准化后查看标准化后的图片类型。 

trans_norm_png = transforms.Normalize([0.5,0.5,0.5,0.5],[0.5,0.5,0.5,0.5])
img_norm = trans_norm_png(img_tensor)
writer.add_image("PNG Normalize",img_norm)

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/786876.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号