图片定量抽取脚本

import os
import random
import shutil


def move_file(file_path, tar_path):
    # 3.填写要复制关键词的list.txt
    keyword_list = open("list.txt", 'r', encoding='utf-8')
    lines = keyword_list.readlines()
    keyword_list.close()
    for keyword in lines:
        keyword = keyword.strip()
        print(keyword)
        save_path = tar_path + keyword
        if not os.path.exists(save_path):
            os.makedirs(save_path)
        temp = os.listdir(file_path + keyword)
        # print(temp)
        # 4.填写要复制的图片数量
        n = int(input("请输入要复制的图片数量："))
        for index, v in enumerate(temp):
            if index < n:
                shutil.copy(file_path + keyword + "\\" + v, save_path + "\\" + v)
                print(v)


if __name__ == '__main__':
    # 1.填写源图片文件夹路径
    # file_dir = input("填写源图片文件夹路径(./test/)：")
    file_dir = r"X:/phenas项目/数据模块/data-clean/0007_校准数据/校准数据/"
    # 2.填写移动到新的文件夹路径
    # tar_dir = input("填写移动到新的文件夹路径(./result/)：")
    tar_dir = r'X:/phenas项目/智能相册项目/同phenas类/'
    move_file(file_dir, tar_dir)