pytorch transforms.ToTensor和transforms.Normalize

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

pytorch transforms.ToTensor和transforms.Normalize

秃头小苏   2022-05-28 我要评论

transforms.ToTensor

最近看pytorch时,遇到了对图像数据的归一化,如下图所示:

image-20220416115017669

该怎么理解这串代码呢?我们一句一句的来看,先看transforms.ToTensor(),我们可以先转到官方给的定义,如下图所示:

image-20220416115331930

大概的意思就是说,transforms.ToTensor()可以将PIL和numpy格式的数据从[0,255]范围转换到[0,1] ,具体做法其实就是将原始数据除以255。另外原始数据的shape是(H x W x C),通过transforms.ToTensor()后shape会变为(C x H x W)。这样说我觉得大家应该也是能理解的,这部分并不难,但想着还是用一些例子来加深大家的映像

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们