添加状态分类和液面分割
This commit is contained in:
@ -80,8 +80,8 @@ def process_images_in_folder(input_folder, output_folder):
|
||||
|
||||
# ================ 使用示例 ================
|
||||
if __name__ == "__main__":
|
||||
folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10"
|
||||
output_folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10"
|
||||
folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f15"
|
||||
output_folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f15"
|
||||
|
||||
if not os.path.exists(folder):
|
||||
print("❌ 输入文件夹不存在!")
|
||||
|
||||
@ -106,7 +106,7 @@ def delete_gray_images(folder_path, extensions=None, dry_run=False):
|
||||
|
||||
|
||||
# ================== 用户配置 ==================
|
||||
FOLDER_PATH = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10" # 修改为你的图片文件夹
|
||||
FOLDER_PATH = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f15" # 修改为你的图片文件夹
|
||||
DRY_RUN = False # 先设为 True 测试,确认无误后再改为 False
|
||||
|
||||
# ================== 执行 ==================
|
||||
|
||||
@ -91,7 +91,7 @@ def delete_similar_consecutive_images(folder_path, threshold=0.95, extensions=No
|
||||
|
||||
|
||||
# ================== 用户配置 ==================
|
||||
FOLDER_PATH = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10" # 修改为你的图片文件夹路径
|
||||
FOLDER_PATH = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f15" # 修改为你的图片文件夹路径
|
||||
THRESHOLD = 0.90 # SSIM 阈值
|
||||
|
||||
# ================== 执行 ==================
|
||||
|
||||
Reference in New Issue
Block a user