瀏覽代碼

修改缺陷分数修改的保存方法

AnlaAnla 2 周之前
父節點
當前提交
e47f8ce501
共有 2 個文件被更改,包括 7 次插入5 次删除
  1. 1 1
      Test/test01.py
  2. 6 4
      app/utils/score_inference/CardScorer.py

+ 1 - 1
Test/test01.py

@@ -219,4 +219,4 @@ result = np.array([
 ], dtype=np.int32).tolist()
 
 print(f"Bounding Rect (x,y,w,h): {x}, {y}, {w}, {h}")
-print(f"Result (4 points):\n{result}")
+print(f"Result (4 points):\n{result}")

+ 6 - 4
app/utils/score_inference/CardScorer.py

@@ -100,14 +100,16 @@ class CardScorer:
                     defect["new_score"] = None
                 elif defect.get("new_score") is None:
                     # 初次修改
-                    if defect["score"] != the_score:
-                        logger.info(f"初次修改 -> new_score: {the_score} (原score: {defect['score']})")
+                    # if defect["score"] != the_score:
+                    #     logger.info(f"初次修改 -> new_score: {the_score} (原score: {defect['score']})")
+                    if defect.get("edit_type") == "edit":
                         defect["new_score"] = the_score
 
                 elif "score" in defect and defect["new_score"] is not None:
                     # 多次修改
-                    if defect["new_score"] != the_score:
-                        defect["score"] = defect["new_score"]
+                    # if defect["new_score"] != the_score:
+                    #     defect["score"] = defect["new_score"]
+                    if defect.get("edit_type") == "edit":
                         defect["new_score"] = the_score
                 else:
                     defect['score'] = the_score