#! /usr/bin/env python
# -*- coding: utf-8 -*-#
# -------------------------------------------------------------------------------
# Name: 图片模糊度计算
# Author: yunhgu
# Date: 2021/8/20 11:01
# Description:
# -------------------------------------------------------------------------------
import shutil
from traceback import format_exc
import cv2
from pathlib import Path
from PIL import Image
import numpy as np
def ImageToMatrix(file):
# 读取图片
im = Image.open(file)
# 显示图片
# im.show()
width, height = im.size
im = im.convert("L")
data = im.getdata()
data = np.matrix(data, dtype='float') / 255.0
new_data = np.reshape(data, (height, width))
return new_data
def Brenner(img):
x, y = img.shape
D = 0
for i in range(x - 2):
for j in range(y - 2):
D += (img[i + 2, j] - img[i, j]) ** 2
return D
def variance_of_laplacian(image):
return cv2.Laplacian(image, cv2.CV_64F).var()
def main(input_path, output_path):
for file in Path(input_path).rglob("*.jpg"):
# 拉普拉斯算子
image = cv2.imdecode(np.fromfile(file, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
fm = variance_of_laplacian(gray)
# Brenner 检测
frame = ImageToMatrix(file)
score = Brenner(frame)
print(fm, score)
# result = Path(output_path).joinpath("no_clear")
# if fm >= 90:
# result = Path(output_path).joinpath("90")
# elif fm >= 80:
# result = Path(output_path).joinpath("80")
# elif fm >= 70:
# result = Path(output_path).joinpath("70")
# elif fm >= 60:
# result = Path(output_path).joinpath("60")
# elif fm >= 50:
# result = Path(output_path).joinpath("50")
# elif fm >= 40:
# result = Path(output_path).joinpath("40")
# elif fm >= 30:
# result = Path(output_path).joinpath("30")
# elif fm >= 20:
# result = Path(output_path).joinpath("20")
# elif fm >= 10:
# result = Path(output_path).joinpath("10")
#
# result.mkdir(parents=True, exist_ok=True)
# shutil.copy(file, result)
if __name__ == '__main__':
print("Start...")
# input_Folder = input("请输入源文件夹:")
# output_Folder = input("请输入结果文件夹:")
input_Folder = r"F:pythonProject图片模糊度计算color"
output_Folder = r"F:pythonProject图片模糊度计算
esult"
try:
main(input_Folder, output_Folder)
print("finished")
except Exception as err:
print(f"程序运行失败!!!请联系数据处理中心:{err}")
print(format_exc())
input("按任意键盘退出!!!")