Bläddra i källkod

修改格式类型

AnlaAnla 3 månader sedan
förälder
incheckning
3b365de75b

+ 1 - 0
app/services/score_service.py

@@ -42,6 +42,7 @@ class ScoreService:
                 return {}
 
         logger.info("模型推理结束, 开始计算分数")
+
         if score_type == 'front_corner_edge' or score_type == 'front_face':
             center_score_data = card_scorer.calculate_centering_score('front', center_data, True)
         else:

+ 3 - 3
app/utils/json_data_formate.py

@@ -3,7 +3,7 @@ import json
 
 def formate_center_data(center_result, inner_data: dict, outer_data: dict):
     data = {
-        "center_result": {
+        "box_result": {
             "center_inference": {
                 "angel_diff": center_result[2],
                 "center_left": center_result[0][0],
@@ -13,8 +13,8 @@ def formate_center_data(center_result, inner_data: dict, outer_data: dict):
             }
         }
     }
-    data['center_result']['inner_box'] = inner_data
-    data['center_result']['outer_box'] = outer_data
+    data['box_result']['inner_box'] = inner_data
+    data['box_result']['outer_box'] = outer_data
     return data
 
 

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

@@ -123,10 +123,10 @@ class CardScorer:
         rules = centering_config['rules']
         coefficients = centering_config['coefficients']
 
-        center_left = center_data['center_result']['center_inference']['center_left']
-        center_right = center_data['center_result']['center_inference']['center_right']
-        center_top = center_data['center_result']['center_inference']['center_top']
-        center_bottom = center_data['center_result']['center_inference']['center_bottom']
+        center_left = center_data['box_result']['center_inference']['center_left']
+        center_right = center_data['box_result']['center_inference']['center_right']
+        center_top = center_data['box_result']['center_inference']['center_top']
+        center_bottom = center_data['box_result']['center_inference']['center_bottom']
 
         # 将比例转换为用于查找规则的单个最大值
         h_lookup_val = max(center_left, center_right)