以图片文件举例:
1、需要准备的东西:
a.含有要提取的图片的文件名的txt文件:
b.总的图片文件
c.创建保存复制的图片的文件夹
2、代码如下:
# -*- coding: utf-8 -*-
import shutil
def objFileName():
"""
生成文件名列表
"""
# 1.存有 需要复制的文件的文件名的txt文件路径
local_file_name_list = r'D:/数据集/dataset/SeaShips(7000)/ImageSets/Main/train.txt'
obj_name_list = []
for i in open(local_file_name_list, 'r'):
obj_name_list.append(i.replace('\n', ''))
return obj_name_list
def copy_img():
"""
复制、重命名、粘贴文件
"""
# 2.指定含有全部文件的文件夹路径
local_img_name = r'D:/数据集/dataset/SeaShips(7000)/images'
# 3.指定存放复制的文件的文件夹路径
path = r'D:/数据集/dataset/SeaShips(7000)/images/train_images'
for i in objFileName():
new_obj_name = i + '.jpg' # 4.改成文件的格式。如图片用.jpg
shutil.copy(local_img_name + '/' + new_obj_name, path + '/' + new_obj_name)
if __name__ == '__main__':
copy_img()
注:代码中注释的1.~4.需要改成自己的。
如何生成这里需要的txt文件见另一篇:http://t.csdn.cn/luAZp