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