# -*- coding: utf-8 -*-
# from mailbox import linesep
import os
import shutil
# import cv2
path0 = r'\testjyz401\labels'#groud truth为txt文件夹
path = r"\jyztest_7class1280_conf0.25_401\images" # jpg图片和对应的生成结果的txt标注文件,放在一起
path3 = r"\jyztest_7class1280_conf0.25_401\2" # 裁剪出来的小图保存的根目录
img_total = []
txt_total = []
file = os.listdir(path)
for filename in file:
first,last = os.path.splitext(filename)
if last == ".jpg": # 图片的后缀名
img_total.append(first)
#print(img_total)
else:
txt_total.append(first)
for img_ in img_total:
if img_ in txt_total:
filename_txt = img_+".txt" # txt的后缀名
filename_img = img_+".jpg" # 图片的后缀名
# print('filename_img:', filename_img)
path1 = os.path.join(path,filename_txt)
pathimg = os.path.join(path,filename_img)
file1 = open(path1)
lines1 = len(file1.readlines())
file0 = open(os.path.join(path0,filename_txt))
lines0 = len(file0.readlines())
if lines1 !=lines0:
print(lines0)
print(lines1)
print('-'*20)
shutil.copy(pathimg, os.path.join(path3,filename_img))
python3挑出检测错误的图片
最新推荐文章于 2023-12-13 18:48:51 发布