56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
import os
|
|||
|
|
import cv2
|
|||
|
|
import shutil
|
|||
|
|
|
|||
|
|
# ====================== 用户配置 ======================
|
|||
|
|
SOURCE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/zjdata15/2" # 原图片文件夹
|
|||
|
|
OUTPUT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/zjdata15/2" # 模糊图输出文件夹
|
|||
|
|
BLUR_THRESHOLD = 180 # 模糊阈值，越大越严格（100~300）
|
|||
|
|
# ======================================================
|
|||
|
|
|
|||
|
|
IMG_EXT = [".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".webp"]
|
|||
|
|
|
|||
|
|
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|||
|
|
|
|||
|
|
def is_blurry(image_path, threshold=150):
|
|||
|
|
"""使用拉普拉斯方差检测是否模糊"""
|
|||
|
|
img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
|
|||
|
|
|
|||
|
|
if img is None:
|
|||
|
|
return False # 读取失败跳过
|
|||
|
|
|
|||
|
|
variance = cv2.Laplacian(img, cv2.CV_64F).var()
|
|||
|
|
|
|||
|
|
return variance < threshold, variance
|
|||
|
|
|
|||
|
|
|
|||
|
|
def scan_and_filter():
|
|||
|
|
total = 0
|
|||
|
|
bad = 0
|
|||
|
|
|
|||
|
|
for filename in os.listdir(SOURCE_DIR):
|
|||
|
|
if not any(filename.lower().endswith(ext) for ext in IMG_EXT):
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
img_path = os.path.join(SOURCE_DIR, filename)
|
|||
|
|
total += 1
|
|||
|
|
|
|||
|
|
is_bad, score = is_blurry(img_path, BLUR_THRESHOLD)
|
|||
|
|
|
|||
|
|
if is_bad:
|
|||
|
|
bad += 1
|
|||
|
|
shutil.copy(img_path, os.path.join(OUTPUT_DIR, filename))
|
|||
|
|
print(f"[低质量] {filename} — 清晰度评分: {score:.2f}")
|
|||
|
|
else:
|
|||
|
|
print(f"[清晰] {filename} — 清晰度评分: {score:.2f}")
|
|||
|
|
|
|||
|
|
print("\n============================")
|
|||
|
|
print(f"扫描总数: {total}")
|
|||
|
|
print(f"检测为模糊/马赛克: {bad}")
|
|||
|
|
print(f"已保存到: {OUTPUT_DIR}")
|
|||
|
|
print("============================\n")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
scan_and_filter()
|