TORCHVISION
该包主要由3个子包组成,分别是:torchvision.datasets
、torchvision.models
、torchvision.transforms
。
主要学习torchvision.transforms
,包括常用的图像操作,例如:随机切割,旋转,数据类型转换,图像到tensor ,numpy 数组到tensor , tensor 到 图像等。
transforms.Compose
可以组合几个变换连续一起操作
比如
transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),
])