2023-12-15 20:31:28 +00:00
|
|
|
"""Functions for assessing image blurriness"""
|
|
|
|
|
|
2023-01-16 22:45:17 +00:00
|
|
|
import cv2
|
|
|
|
|
|
2023-12-15 21:40:10 +00:00
|
|
|
from imaginairy.utils.img_utils import pillow_img_to_opencv_img
|
2023-01-16 22:45:17 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def calculate_blurriness_level(img):
|
|
|
|
|
img = pillow_img_to_opencv_img(img)
|
|
|
|
|
|
|
|
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
|
|
|
|
sharpness = cv2.Laplacian(gray, cv2.CV_64F).var()
|
|
|
|
|
sharpness = max(sharpness, 0.000001)
|
|
|
|
|
bluriness = 1 / sharpness
|
|
|
|
|
return bluriness
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def is_blurry(img, threshold=0.91):
|
|
|
|
|
return calculate_blurriness_level(img) > threshold
|
