79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
import os
|
||
from pathlib import Path
|
||
|
||
# ==================== 配置 ====================
|
||
FOLDER_PATH = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/11c" # 替换为你的数据目录
|
||
IMG_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.bmp', '.tif', '.tiff', '.webp'}
|
||
DRY_RUN = False # True: 仅预览；False: 真实删除
|
||
# ==============================================
|
||
|
||
def main():
|
||
folder = Path(FOLDER_PATH)
|
||
if not folder.exists():
|
||
print(f"❌ 目录不存在: {folder}")
|
||
return
|
||
|
||
# 获取所有图片（stem -> full path）
|
||
img_files = {}
|
||
for ext in IMG_EXTENSIONS:
|
||
for f in folder.glob(f"*{ext}"):
|
||
img_files[f.stem] = f
|
||
|
||
# 获取所有 txt 文件（stem -> full path）
|
||
txt_files = {}
|
||
for f in folder.glob("*.txt"):
|
||
txt_files[f.stem] = f
|
||
|
||
all_stems = set(img_files.keys()) | set(txt_files.keys())
|
||
|
||
to_delete_imgs = []
|
||
to_delete_txts = []
|
||
|
||
for stem in sorted(all_stems):
|
||
has_img = stem in img_files
|
||
has_txt = stem in txt_files
|
||
|
||
if not has_img and has_txt:
|
||
to_delete_txts.append(txt_files[stem])
|
||
elif has_img and not has_txt:
|
||
to_delete_imgs.append(img_files[stem])
|
||
# 如果都有，保留；如果都无（理论上不会），跳过
|
||
|
||
# 输出结果
|
||
print("🔍 扫描完成。")
|
||
print(f"🖼️ 图片数量: {len(img_files)}")
|
||
print(f"📄 TXT 数量 : {len(txt_files)}")
|
||
print()
|
||
|
||
if to_delete_imgs:
|
||
print("🗑️ 将删除无对应 TXT 的图片:")
|
||
for f in to_delete_imgs:
|
||
print(f" - {f}")
|
||
else:
|
||
print("✅ 所有图片都有对应的 TXT")
|
||
|
||
if to_delete_txts:
|
||
print("\n🗑️ 将删除无对应图片的 TXT:")
|
||
for f in to_delete_txts:
|
||
print(f" - {f}")
|
||
else:
|
||
print("✅ 所有 TXT 都有对应的图片")
|
||
|
||
total_to_delete = len(to_delete_imgs) + len(to_delete_txts)
|
||
print(f"\n📌 总共将删除 {total_to_delete} 个文件")
|
||
|
||
if DRY_RUN:
|
||
print("\n💡 当前为预览模式 (DRY_RUN=True)，未执行真实删除。")
|
||
print(" 如需真实删除，请将 DRY_RUN 改为 False 并重新运行。")
|
||
else:
|
||
print("\n⚠️ 正在执行删除...")
|
||
for f in to_delete_imgs + to_delete_txts:
|
||
try:
|
||
f.unlink()
|
||
print(f"✅ 已删除: {f}")
|
||
except Exception as e:
|
||
print(f"❌ 删除失败: {f} - {e}")
|
||
print("✅ 清理完成！")
|
||
|
||
if __name__ == "__main__":
|
||
main() |