Skip to content

undergroundCAT/study-on-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

study-on-pytorch

关于torchvision.transforms.ToTensor的问题

在官方文档中,class torchvision.transforms.ToTensor是把一个取值范围在0-255的np.array转化成0-1之间的torch.FloadTensor。

这里可以将数组自动归一化,官方的例子是:

data = np.random.randint(0, 255, size=300) img = data.reshape(10,10,3) print(img.shape) img_tensor = transforms.ToTensor()(img) # 转换成tensor print(img_tensor)

image

明显这里并没有将想要转化的数据进行归一化的转化。

这里将np数组转成uint8的数据类型就可以实现归一化。

image

image

若数据类型是其他,均无法进行归一化。

About

关于torchvision.transforms.ToTensor的问题

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published