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

torchvision 笔记:transforms.Normalize()

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

torchvision 笔记:transforms.Normalize()

        一般和transforms.ToTensor()搭配使用

        作用就是先将输入归一化到(0,1)【transforms.ToTensor()】,再使用公式"(x-mean)/std",将每个元素分布到(-1,1)

   很多CV的代码中,是这样使用这一条语句的:

torchvision.transforms.Normalize(
    mean=[0.485, 0.456, 0.406], 
    std=[0.229, 0.224, 0.225])

这一组参数是从ImageNet数据集中获得的

在 torchvision 笔记:ToTensor()_UQI-LIUWJ的博客-CSDN博客的代码基础上我们进行修改

ToTensor 中的代码:
from PIL import Image
from torchvision import transforms, utils
a=Image.open(b+'img/00000.jpg')
a

 

y=transforms.ToTensor()
a=y(a)
a
'''
tensor([[[0.9255, 0.9255, 0.9255,  ..., 0.9176, 0.9176, 0.9176],
         [0.9255, 0.9255, 0.9255,  ..., 0.9176, 0.9176, 0.9176],
         [0.9255, 0.9255, 0.9255,  ..., 0.9176, 0.9176, 0.9176],
         ...,
         [0.7882, 0.7882, 0.7882,  ..., 0.7922, 0.7922, 0.7922],
         [0.7882, 0.7882, 0.7882,  ..., 0.7922, 0.7922, 0.7922],
         [0.7882, 0.7882, 0.7882,  ..., 0.7922, 0.7922, 0.7922]],

        [[0.9255, 0.9255, 0.9255,  ..., 0.9216, 0.9216, 0.9216],
         [0.9255, 0.9255, 0.9255,  ..., 0.9216, 0.9216, 0.9216],
         [0.9255, 0.9255, 0.9255,  ..., 0.9216, 0.9216, 0.9216],
         ...,
         [0.7961, 0.7961, 0.7961,  ..., 0.7922, 0.7922, 0.7922],
         [0.7961, 0.7961, 0.7961,  ..., 0.7922, 0.7922, 0.7922],
         [0.7961, 0.7961, 0.7961,  ..., 0.7922, 0.7922, 0.7922]],

        [[0.9255, 0.9255, 0.9255,  ..., 0.9294, 0.9294, 0.9294],
         [0.9255, 0.9255, 0.9255,  ..., 0.9294, 0.9294, 0.9294],
         [0.9255, 0.9255, 0.9255,  ..., 0.9294, 0.9294, 0.9294],
         ...,
         [0.7922, 0.7922, 0.7922,  ..., 0.8000, 0.8000, 0.8000],
         [0.7922, 0.7922, 0.7922,  ..., 0.8000, 0.8000, 0.8000],
         [0.7922, 0.7922, 0.7922,  ..., 0.8000, 0.8000, 0.8000]]])
'''
 Normalize的代码
z=transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
a=z(a)
a
'''
tensor([[[1.9235, 1.9235, 1.9235,  ..., 1.8893, 1.8893, 1.8893],
         [1.9235, 1.9235, 1.9235,  ..., 1.8893, 1.8893, 1.8893],
         [1.9235, 1.9235, 1.9235,  ..., 1.8893, 1.8893, 1.8893],
         ...,
         [1.3242, 1.3242, 1.3242,  ..., 1.3413, 1.3413, 1.3413],
         [1.3242, 1.3242, 1.3242,  ..., 1.3413, 1.3413, 1.3413],
         [1.3242, 1.3242, 1.3242,  ..., 1.3413, 1.3413, 1.3413]],

        [[2.0959, 2.0959, 2.0959,  ..., 2.0784, 2.0784, 2.0784],
         [2.0959, 2.0959, 2.0959,  ..., 2.0784, 2.0784, 2.0784],
         [2.0959, 2.0959, 2.0959,  ..., 2.0784, 2.0784, 2.0784],
         ...,
         [1.5182, 1.5182, 1.5182,  ..., 1.5007, 1.5007, 1.5007],
         [1.5182, 1.5182, 1.5182,  ..., 1.5007, 1.5007, 1.5007],
         [1.5182, 1.5182, 1.5182,  ..., 1.5007, 1.5007, 1.5007]],

        [[2.3088, 2.3088, 2.3088,  ..., 2.3263, 2.3263, 2.3263],
         [2.3088, 2.3088, 2.3088,  ..., 2.3263, 2.3263, 2.3263],
         [2.3088, 2.3088, 2.3088,  ..., 2.3263, 2.3263, 2.3263],
         ...,
         [1.7163, 1.7163, 1.7163,  ..., 1.7511, 1.7511, 1.7511],
         [1.7163, 1.7163, 1.7163,  ..., 1.7511, 1.7511, 1.7511],
         [1.7163, 1.7163, 1.7163,  ..., 1.7511, 1.7511, 1.7511]]])
'''

将tensor反变换回图片,则有

 

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

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

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