需求: validationImages.csv文件是存储验证集数据名称和类别信息(labels)的文件, 要生成一个label和类别名一一对应且正序排列的json文件,代码如下: labels_dict = {} with open(os.path.join(os.getcwd(), 'validationImages.csv')) as f: reader = csv.reader(f) for row in reader: if row[0].split(os.sep)[-2][4:] n
整理一下看到的自定义数据读取的方法,较好的有一下三篇文章, 其实自定义的方法就是把现有数据集的train和test分别用 含有图像路径与label的list返回就好了,所以需要根据数据集随机应变. 所有图片都在一个文件夹1 之前刚开始用的时候,写Dataloader遇到不少坑.网上有一些教程 分为all images in one folder 和 each class one folder.后面的那种写的人比较多,我写一下前面的这种,程式化的东西,每次不同的任务改几个参数就好. 等训练的时候写