判断图片清晰度

util/image_util.py

@@ -251,3 +251,20 @@ def combined(img1, img2):
     combined_img[:height1, :width1] = img1
     combined_img[:height2, width1:width1 + width2] = img2
     return combined_img
+
+
+def parse_clarity(image):
+    """
+    判断图片清晰度
+    :param image: 图片NumPy数组或文件路径
+    :return: 判断结果及置信度
+    """
+    clarity_result = [1, 0]
+    model = PaddleClas(inference_model_dir=r"model/clas/clarity_assessment", use_gpu=False)
+    clas_result = model.predict(input_data=image)
+    try:
+        clas_result = next(clas_result)[0]
+        clarity_result = [clas_result["class_ids"][0], clas_result["scores"][0]]
+    except Exception as e:
+        logging.error("获取图片清晰度失败", exc_info=e)
+    return clarity_result