在做項(xiàng)目時(shí)遇到需要標(biāo)記數(shù)據(jù)集里面的若干圖片數(shù)據(jù),作為程序員,為避免手動(dòng)一張一張的篩選,所以寫(xiě)了這個(gè)Python腳本實(shí)現(xiàn)。
Python腳本如下:
# from PIL import Imageimport csvimport osimport 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/項(xiàng)目組/text_detect_label_data/China_SameBrowser"): # print(image[:-4]) if image[:-4] in pickImageNames: # pickImage = Image.open((r"C:/Users/Administrator.PC-201708272051/Desktop/項(xiàng)目組/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/項(xiàng)目組/text_detect_label_data/China_SameBrowser/" + image newname= r"C:/Users/Administrator.PC-201708272051/Desktop/labeldata/" + image shutil.copyfile(oldname,newname) # readImageName()pickImg()
以上這篇用Python實(shí)現(xiàn)篩選文件腳本的方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持武林站長(zhǎng)站。
新聞熱點(diǎn)
疑難解答
圖片精選