torchvision 作用

1. 提供主流的model,和常用数据集

2. 对 torch.utils.data.Dataset进行扩充,针对不同类别图像放入不同文件夹的数据进行读取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子类。

3.提供transforms

原文地址:https://www.cnblogs.com/fanhaha/p/8046166.html