码迷,mamicode.com
首页 > 其他好文 > 详细

torchvision 作用

时间:2017-12-16 13:20:09      阅读:182      评论:0      收藏:0      [点我收藏+]

标签:rms   datasets   不同类   tor   form   ase   post   常用   不同   

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

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

3.提供transforms

torchvision 作用

标签:rms   datasets   不同类   tor   form   ase   post   常用   不同   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!