前言
无论是在作分类任务或者是目标检测任务都需要数据集的处理,一种是txt文件保存标签的信息,另一种只有图片如下图的形式,这一步也是学会faster-rcnn的关键点
分为训练和验证的照片
|
每个分类的类别
一种是猫的照片,另一种是狗的照片,这种是自己的数据集,其实官方的数据集也是这样放置的,比如CIFAR10,其中的是有10个文件夹,每个文件夹下是很多张一种数字的照片,正常情况下我们引进官方数据集的写法如下
transform = transforms.Compose([
transforms.RandomHorizontalFlip(), # 在小型数据集上,通过随机水
1