我正在尝试将大型数据集拆分为来自Food101数据集的列车/有效/测试集以进行图像分类
并且数据集的结构是这样的,并且所有图像都在一个文件夹中
’,” …
我似乎对解决方案一切都错了,我不需要移动所有需要更改的图像是通过os模块以所需格式显示图像的路径
以下是执行此操作的代码。假设您有有效列表中的文件名列表
#for valid set v = valid.reshape(15150,) or_fpath = '/content/food-101/images/' #path of original folder cp_fpath = '/content/food101/valid/' #path of destination folder for y in tqdm(v): foldername = y.split('/')[0] img = y.split('/')[1] +'.jpg' ip_path = or_fpath+foldername op_path = cp_fpath+foldername if not os.path.exists(op_path): os.mkdir(op_path) os.rename(os.path.join(ip_path, img), os.path.join(op_path, img))
谢谢!
注意:如果你有更好的答案,请分享谢谢