用Python将文件夹中特定的文件批量批量移动出来
因为在做深度学习的时候要自己标注数据集,数据集中图片和标签文件是一一对应的,文件名一样,只不过后缀不一样。有时候需要从总的数据集中选出一部分出来(图片和标签文件,我这里是.jpg文件和.txt文件),我这里.jpg文件是自己一张一张选出来的,而标签文件的名称和.jpg文件是一样的,只不过是后缀不同,所以就写了一个脚本自动移出来,记录一下,方便以后查看。
在桌面新建一个文件夹
然后运行代码。
import os
import shutil
train_list = []
for i in os.listdir('C:\\Users\\1\\Desktop\\1\\train'): #遍历train文件夹下的所有jpg文件
i = i.split('.') #将jpg文件的前缀和后缀分开
i = i[0] + '.txt' #将.jpg文件名改为.txt
a = ('C:\\Users\\1\\Desktop\\1\\labels\\' + i)
shutil.move(a, 'C:\\Users\\1\\Desktop\\1\\train_labels') #将label文件夹对应的.txt文件移动到train_labels文件夹