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

docker-compose.yml

@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.14.17
+  image: fcb_photo_review:1.15.0
   restart: always
 
 x-review:

photo_review/auto_photo_review.py

@@ -258,6 +258,18 @@ def information_extraction(ie, phrecs, identity):
         session = MysqlSession()
         session.add_all(zx_ie_results)
         session.commit()
+
+        # 添加清晰度测试
+        if not better_image:
+            # 替换后图片默认清晰
+            clarity_result = image_util.parse_clarity(image)
+            unsharp_flag = 0 if (clarity_result[0] == 0 and clarity_result[1] >= 0.8) else 1
+            update_clarity = (update(ZxPhrec).where(ZxPhrec.pk_phrec == phrec.pk_phrec).values(
+                cfjaddress2=json.dumps(clarity_result),
+                unsharp_flag=unsharp_flag,
+            ))
+            session.execute(update_clarity)
+            session.commit()
         session.close()
 
     result['ocr_text'] = ocr_text

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]