pytorch之transforms(numpy数据类型转化为tensor,归一化、resize)

最新推荐文章于 2024-07-21 21:30:08 发布

_腾飞

最新推荐文章于 2024-07-21 21:30:08 发布

阅读量988

点赞数

分类专栏： # 基础包文章标签： numpy pytorch python

本文链接：https://blog.csdn.net/helloworld_Fly/article/details/125861762

版权

基础包专栏收录该内容

39 篇文章 1 订阅

订阅专栏

pytorch transform常用格式

from torchvision import transforms
tensor = transforms.Tensor() #初始化对象
#img_or_np为“PIL Image”或者"numpy.",即你要转化的对象
data = tensor(img_or_np) 
trans_resize = transforms.Resize(512,512)
data = trans_resize(data)
#下面将数据data归一化，计算方法为：image=(image-mean)/std
'''
其中mean和std分别通过(0.5,0.5,0.5)和(0.5,0.5,0.5)进行指定。
原来的0-1最小值0则变成(0-0.5)/0.5=-1，而最大值1则变成(1-0.5)/0.5=1.
'''
trans_norm = transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])
img_normal = trans_norm(data)