2017年9月12日
pytorch学习：准备自己的图片数据
摘要： 图片数据一般有两种情况： 1、所有图片放在一个文件夹内，另外有一个txt文件显示标签。 2、不同类别的图片放在不同的文件夹内，文件夹就是图片的类别。 针对这两种不同的情况，数据集的准备也不相同，第一种情况可以自定义一个Dataset，第二种情况直接调用torchvision.datasets.Ima 阅读全文
posted @ 2017-09-12 22:17 denny402 阅读(32702) 评论(7) 推荐(2) 编辑