添加图片清晰度测试，保存结果对照

2025-04-01 15:11:40 +08:00
parent 3e9c0c99b9
commit 5287df4959
3 changed files with 14 additions and 2 deletions
--- a/util/image_util.py
+++ b/util/image_util.py
@@ -260,7 +260,7 @@ def parse_clarity(image):
    :return: 判断结果及置信度
    """
    clarity_result = [1, 0]
-    model = PaddleClas(inference_model_dir=r"model/clas/clarity_assessment", use_gpu=False)
+    model = PaddleClas(inference_model_dir=r"model/clas/clarity_assessment", use_gpu=True)
    clas_result = model.predict(input_data=image)
    try:
        clas_result = next(clas_result)[0]