import os
import shutil
from PIL import Image
import numpy as np
#判定两个文件夹内的文件是否相同
file_path1= r'D:\数据集\image/' #已知 内容较少的文件夹
file_path2 = r'D:\数据集\yolo_labels2/'
f1 = []
f2 = []
for inputfile in os.listdir(file_path1):
if os.path.splitext(inputfile)[1] == '.jpg':#判断是否为csv文件
fileName = os.path.splitext(inputfile)[0]
f1.append(fileName) #将“文件名”装入列表
for inputfile1 in os.listdir(file_path2):
if os.path.splitext(inputfile1)[1] == '.txt':#判断是否为csv文件
fileName = os.path.splitext(inputfile1)[0]
f2.append(fileName) #将“文件名”装入列表
##将两个文件夹内的文件名不同的提出来
i=0
for filename2 in f1:
f = []
if filename2 not in f2:
i=i+1
shutil.move(r"D:\数据集\image/{}.jpg".format(filename2), r"D:\数据集\image_save" ,) #文件夹需要创建
else:
print("已经存在")
print("已经有{}个图片没有对应的txt文件".format(i))