视觉工作包torchvision主要包含一下三个部分:
- models :提供深度学习中各种经典的模型及网络结合,包括AlexNet、VGG系列、ResNet系列、Inception系列等
- dataset :提供常用的数据集加载,继承torch.utis.data.Dataset
- transforms :提供常用的数据预处理操作
torchvision还提供了两个常用函数
make_grid,它将多张图片拼接在一个网格中
save_img,它将tensor保存成图片
视觉工作包torchvision主要包含一下三个部分:
make_grid,它将多张图片拼接在一个网格中
save_img,它将tensor保存成图片