transforms
Torchvision 中的 transforms
主要用于图像变换。
from torchvision import transforms
源码为 transform.py, 就是一个 python 文件.
- ToTensor:
tensor_trans = transforms.ToTensor()
# 创建自己的工具tensor_img = tensor_trans(img)
# 用自己的工具传递参数对图像进行操作- ?能一步写完吗
tensor_img = transforms.ToTensor()(img)
,
PIL(Python Image Library)是python的第三方图像处理库
- 随机裁剪
数据集:
dataset
dataloader
Containers:神经网络骨架的搭建
torch.nn