import os
import cv2
import shutil
import sys
 
# 模糊影像檢測函數(shù)，閾值默認為0.07
def blurImagesDetection(folder_path, thres=0.07):
    # 新建一個用于存放模糊影像的文件夾
    blurImageDirPath = os.getcwd() + "/blurImages"
    if not os.path.exists(blurImageDirPath):
        os.mkdir(blurImageDirPath)
    # 獲取影像文件夾中的影像名列表
    imageNameList = os.listdir(folder_path)
    for imageName in imageNameList:
        # 得到影像路徑
        imagePath = os.path.join(folder_path, imageName)
        # 讀取影像為灰度圖
        img = cv2.imread(imagePath, 0)
        # 縮小影像，加快處理速度
        tiny_img = cv2.resize(img, (400, 300), fx=0, fy=0)
        # 獲取影像尺寸
        width, height = tiny_img.shape
        # 計算影像的模糊程度
        blurness = cv2.Laplacian(tiny_img, cv2.CV_64F).var() / (width * height)
        # 如果影像模糊程度小于閾值就將其移動到存放模糊影像的文件夾中
        if blurness < thres:
            print(imageName + "  bulrness:%f   模糊" % (blurness))
            blurImagePath = os.path.join(blurImageDirPath, imageName)
            shutil.move(imagePath, blurImagePath)
        else:
            print(imageName + "  blurness:%f   不模糊" % (blurness))
if __name__ == '__main__':
    # 指定要處理的文件夾路徑，sys.argv[1]為第一個參數(shù)
    folder_path = os.getcwd()+'/'+sys.argv[1]
    # 調(diào)用函數(shù)
    blurImagesDetection(folder_path)