torchvision 作用

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

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

3.提供transforms

posted @   hahahaf  阅读(2486)  评论(0)  编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示