CheckCardBoxAndDefectServer/app/utils/json_data_formate.py

import json


def formate_center_data(center_result, inner_data: dict, outer_data: dict):
    data = {
        "center_result": {
            "center_inference": {
                "angel_diff": center_result[2],
                "center_left": center_result[0][0],
                "center_right": center_result[0][1],
                "center_top": center_result[1][0],
                "center_bottom": center_result[1][1]
            }
        }
    }
    data['center_result']['inner_box'] = inner_data
    data['center_result']['outer_box'] = outer_data
    return data


def formate_face_data(area_json: dict):
    for defect in area_json['defects']:
        # 添加新标签
        defect['defect_type'] = "face"
    return area_json


def formate_one_card_result(center_result: dict, defect_result: dict):
    data = {
        "img_id": 2,
        "img_url": "https://123.jpg",
        "result": {
            "center_result": center_result,
            "defect_result": defect_result
        }
    }
    return data

# if __name__ == '__main__':
#     json_path = r"C:\Code\ML\Project\CheckCardBoxAndDefectServer\_temp_work\pokemon_front_face_no_reflect_defect-area_result.json"
#     with open(json_path, 'r') as f:
#         area_json = json.load(f)
#     result = formate_face_data(area_json)
#     print(result)