19 lines
737 B
Python
19 lines
737 B
Python
|
import os
|
||
|
|
|
||
|
|
# 设置图片和标签文件夹路径
|
||
|
|
image_dir = r"g:\smoke\smoke_old\images" # 图片文件夹路径
|
||
|
|
label_dir = r"g:\smoke\smoke_old\labels" # 标签文件夹路径
|
||
|
|
|
||
|
|
# 获取图片文件名（不包括扩展名）
|
||
|
|
image_files = {os.path.splitext(f)[0] for f in os.listdir(image_dir) if f.lower().endswith(('.jpg', '.png', '.jpeg'))}
|
||
|
|
|
||
|
|
# 遍历标签文件夹
|
||
|
|
for label_file in os.listdir(label_dir):
|
||
|
|
label_name, ext = os.path.splitext(label_file)
|
||
|
|
|
||
|
|
if ext.lower() == '.txt' and label_name not in image_files:
|
||
|
|
# 如果标签文件没有对应的图片，删除该标签文件
|
||
|
|
label_path = os.path.join(label_dir, label_file)
|
||
|
|
os.remove(label_path)
|
||
|
|
print(f"Deleted: {label_file}")
|