开课吧cv6项目三:使用62类交通标识图片的数据集完成多分类任务
Pytorch
由于相同类别的图片都在同一文件夹中,所以使用torchvision的datasets.ImageFolder来自动生成标签和数据。
原训练集有4572张图片,将结尾为'0000.png'的图片共1548张拆分成验证集,新训练集剩3024张图片。再通过13中方式对拆分后的训练集图片进行增广,得到39312张训练图片。
使用torch.utils.data中的Dataloader读取数据。
'model_40.pth'对2520张测试集图片进行测试,准确率98.2539%。