74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
基于行扫描和统计特征的卡片截取工具
|
||
"""
|
||
import sys
|
||
import os
|
||
|
||
# 添加项目根目录到系统路径
|
||
project_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
sys.path.append(project_root)
|
||
from Util import Kit
|
||
|
||
def clean_directory(dir_path):
|
||
"""
|
||
清理现场：删除目录下除纯数字.jpg以外的所有文件
|
||
"""
|
||
print(f"Cleaning directory: {dir_path}")
|
||
if not os.path.exists(dir_path):
|
||
return
|
||
|
||
for filename in os.listdir(dir_path):
|
||
path = os.path.join(dir_path, filename)
|
||
if not os.path.isfile(path):
|
||
continue
|
||
|
||
name, ext = os.path.splitext(filename)
|
||
# 保留条件：后缀是 .jpg 且文件名是纯数字
|
||
# [修改] 同时保留 _flag.jpg, _vl.jpg 供查看
|
||
# [修改] 保留 .json 文件
|
||
if ext.lower() == ".json":
|
||
continue
|
||
|
||
if ext.lower() == ".jpg" and (name.isdigit() or name.endswith("_flag") or name.endswith("_vl")):
|
||
continue
|
||
|
||
# 清理旧的 _for_vl.jpg
|
||
if name.endswith("_for_vl"):
|
||
pass # Let it fall through to delete
|
||
else:
|
||
# 其他非生成的文件，可能需要保留吗？
|
||
# 这里的逻辑是清理 output 目录，假设该目录下只有生成的图片
|
||
# 如果是原始图片（比如 1.jpg），不能删！
|
||
pass
|
||
|
||
# 原始图片保护 (文件名比较短，通常是 1.jpg, 2.jpg 等)
|
||
if len(name) <= 2 and name.isdigit():
|
||
continue
|
||
|
||
try:
|
||
os.remove(path)
|
||
print(f" Deleted: {filename}")
|
||
except Exception as e:
|
||
print(f" Error deleting {filename}: {e}")
|
||
|
||
def crop_cards(img_path):
|
||
Kit.crop_cards_from_image(img_path)
|
||
|
||
if __name__ == "__main__":
|
||
test_files = [
|
||
r"d:\dsWork\dsProject\dsCrawler\Tools\Images\1.jpg",
|
||
r"d:\dsWork\dsProject\dsCrawler\Tools\Images\2.jpg",
|
||
r"d:\dsWork\dsProject\dsCrawler\Tools\Images\3.jpg",
|
||
r"d:\dsWork\dsProject\dsCrawler\Tools\Images\4.jpg"
|
||
]
|
||
|
||
# 在测试前清理现场
|
||
if test_files:
|
||
# 假设所有图片都在同一个文件夹
|
||
target_dir = os.path.dirname(test_files[0])
|
||
clean_directory(target_dir)
|
||
|
||
for f in test_files:
|
||
crop_cards(f)
|