用Python实现筛选文件脚本

    在做项目时遇到需要标记数据集里面的若干图片数据，作为程序员，为避免手动一张一张的筛选，所以写了这个Python脚本实现。

Python脚本如下：

# from PIL import Image
import csv
import os
import shutil

filename = 'img.txt'

def readImageName():
    with open(filename) as f:
        lines = f.readlines()
        imgnames = []
        for line in lines:
            imgnames.append(line.strip().strip(".jpg")[-4:])
        print(imgnames)
        return imgnames

def pickImg():
    pickImageNames = readImageName()
    # 遍历所有图片集的文件名
    for image in os.listdir(r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser"):
        # print(image[:-4])
        if image[:-4] in pickImageNames:
            # pickImage = Image.open((r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser/%s") % image)
            # pickImage.save((r"C:/Users/Administrator.PC-201708272051/Desktop/labeldata/%s") % image)

            oldname= r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser/" + image
            newname= r"C:/Users/Administrator.PC-201708272051/Desktop/labeldata/" + image
            shutil.copyfile(oldname,newname)

# readImageName()
pickImg()