需要修改的代码
1.class DemoDataset(BaseCVDataset):
#修改路径
self.dataset_dir = "/home/aistudio/dataset"
2.文件存储格式
xujiaqi/1.jpg 1
3.预测，文件修改如下：
data = ['dataset/'+filepath[0].split(" ")[0],'dataset/'+filepath[1].split(" ")[0],'dataset/'+filepath[2].split(" ")[0],'dataset/'+filepath[3].split(" ")[0],'dataset/'+filepath[4].split(" ")[0]]
原创文章 13
获赞 5
访问量 1万+