添加模拟查询

Test/test01.py

@@ -0,0 +1,6 @@
+import json
+
+with open("../temp/test_return.json", 'r') as f:
+    data = json.load(f)
+
+print(data)

app/api/card_inference.py

@@ -4,6 +4,7 @@ from typing import Annotated
 from enum import Enum
 from ..core.config import settings
 from app.services.card_service import CardInferenceService, card_service
+import json
 
 router = APIRouter()
 
@@ -38,3 +39,14 @@ async def card_json_result(
         raise HTTPException(status_code=400, detail=str(e))
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"服务器内部错误: {e}")
+
+
+@router.post("/mock_query")
+async def mock_query(img_id: int):
+    # json_data = {"img_id": img_id}
+    with open("temp/test_return.json", "r") as f:
+        json_data = json.load(f)
+
+    return json_data
+
+

app/core/model_loader.py

@@ -4,6 +4,9 @@ from ..utils.fry_bisenetv2_predictor_V01_250811 import FryBisenetV2Predictor
 
 # 全局的模型预测器字典
 predictors: Dict[str, FryBisenetV2Predictor] = {}
+'''
+predictors['outer_box'] ...
+'''
 
 
 def load_models():

temp/test_return.json

@@ -0,0 +1,1786 @@
+{
+  "img_id": 2,
+  "img_url": "https://archives.bulbagarden.net/media/upload/5/50/WhimsicottUnifiedMinds144.jpg",
+  "inference_result": {
+    "center_inference": {
+      "center_left": -0.1,
+      "center_bottom": 0.2,
+      "inner_box": {
+        "num": 1,
+        "cls": [
+          1
+        ],
+        "names": [
+          "inner_box"
+        ],
+        "conf": 0.9988297375720048,
+        "shapes": [
+          {
+            "class_num": 1,
+            "label": "inner_box",
+            "probability": 0.9988297375720048,
+            "points": [
+              [
+                1594,
+                184
+              ],
+              [
+                1592,
+                186
+              ],
+              [
+                1493,
+                186
+              ],
+              [
+                1491,
+                187
+              ],
+              [
+                1294,
+                187
+              ],
+              [
+                1293,
+                189
+              ],
+              [
+                1178,
+                189
+              ],
+              [
+                1176,
+                190
+              ],
+              [
+                1056,
+                190
+              ],
+              [
+                1054,
+                192
+              ],
+              [
+                958,
+                192
+              ],
+              [
+                957,
+                194
+              ],
+              [
+                790,
+                194
+              ],
+              [
+                789,
+                195
+              ],
+              [
+                590,
+                195
+              ],
+              [
+                589,
+                197
+              ],
+              [
+                445,
+                197
+              ],
+              [
+                443,
+                198
+              ],
+              [
+                440,
+                198
+              ],
+              [
+                440,
+                200
+              ],
+              [
+                437,
+                203
+              ],
+              [
+                437,
+                205
+              ],
+              [
+                432,
+                210
+              ],
+              [
+                432,
+                211
+              ],
+              [
+                430,
+                213
+              ],
+              [
+                430,
+                216
+              ],
+              [
+                429,
+                218
+              ],
+              [
+                429,
+                256
+              ],
+              [
+                430,
+                258
+              ],
+              [
+                430,
+                518
+              ],
+              [
+                432,
+                520
+              ],
+              [
+                432,
+                755
+              ],
+              [
+                434,
+                757
+              ],
+              [
+                434,
+                926
+              ],
+              [
+                435,
+                928
+              ],
+              [
+                435,
+                1050
+              ],
+              [
+                437,
+                1051
+              ],
+              [
+                437,
+                1214
+              ],
+              [
+                438,
+                1216
+              ],
+              [
+                438,
+                1347
+              ],
+              [
+                440,
+                1349
+              ],
+              [
+                440,
+                1547
+              ],
+              [
+                442,
+                1549
+              ],
+              [
+                442,
+                1688
+              ],
+              [
+                443,
+                1690
+              ],
+              [
+                443,
+                1782
+              ],
+              [
+                445,
+                1784
+              ],
+              [
+                445,
+                1808
+              ],
+              [
+                443,
+                1810
+              ],
+              [
+                445,
+                1811
+              ],
+              [
+                445,
+                1834
+              ],
+              [
+                443,
+                1835
+              ],
+              [
+                445,
+                1837
+              ],
+              [
+                445,
+                1926
+              ],
+              [
+                446,
+                1928
+              ],
+              [
+                446,
+                1931
+              ],
+              [
+                448,
+                1933
+              ],
+              [
+                448,
+                1936
+              ],
+              [
+                450,
+                1938
+              ],
+              [
+                450,
+                1939
+              ],
+              [
+                453,
+                1942
+              ],
+              [
+                454,
+                1942
+              ],
+              [
+                456,
+                1944
+              ],
+              [
+                458,
+                1944
+              ],
+              [
+                459,
+                1946
+              ],
+              [
+                462,
+                1946
+              ],
+              [
+                464,
+                1947
+              ],
+              [
+                518,
+                1947
+              ],
+              [
+                520,
+                1946
+              ],
+              [
+                614,
+                1946
+              ],
+              [
+                616,
+                1944
+              ],
+              [
+                766,
+                1944
+              ],
+              [
+                768,
+                1942
+              ],
+              [
+                934,
+                1942
+              ],
+              [
+                936,
+                1941
+              ],
+              [
+                1059,
+                1941
+              ],
+              [
+                1061,
+                1939
+              ],
+              [
+                1165,
+                1939
+              ],
+              [
+                1166,
+                1938
+              ],
+              [
+                1378,
+                1938
+              ],
+              [
+                1379,
+                1936
+              ],
+              [
+                1613,
+                1936
+              ],
+              [
+                1614,
+                1934
+              ],
+              [
+                1630,
+                1934
+              ],
+              [
+                1632,
+                1936
+              ],
+              [
+                1653,
+                1936
+              ],
+              [
+                1654,
+                1934
+              ],
+              [
+                1658,
+                1934
+              ],
+              [
+                1659,
+                1933
+              ],
+              [
+                1661,
+                1933
+              ],
+              [
+                1667,
+                1926
+              ],
+              [
+                1667,
+                1925
+              ],
+              [
+                1669,
+                1923
+              ],
+              [
+                1669,
+                1920
+              ],
+              [
+                1670,
+                1918
+              ],
+              [
+                1670,
+                1914
+              ],
+              [
+                1672,
+                1912
+              ],
+              [
+                1670,
+                1910
+              ],
+              [
+                1670,
+                1646
+              ],
+              [
+                1669,
+                1645
+              ],
+              [
+                1669,
+                1448
+              ],
+              [
+                1667,
+                1446
+              ],
+              [
+                1667,
+                1267
+              ],
+              [
+                1666,
+                1266
+              ],
+              [
+                1666,
+                1074
+              ],
+              [
+                1664,
+                1072
+              ],
+              [
+                1664,
+                962
+              ],
+              [
+                1662,
+                960
+              ],
+              [
+                1662,
+                842
+              ],
+              [
+                1661,
+                840
+              ],
+              [
+                1661,
+                672
+              ],
+              [
+                1659,
+                670
+              ],
+              [
+                1659,
+                507
+              ],
+              [
+                1658,
+                506
+              ],
+              [
+                1658,
+                405
+              ],
+              [
+                1656,
+                403
+              ],
+              [
+                1656,
+                269
+              ],
+              [
+                1654,
+                267
+              ],
+              [
+                1654,
+                202
+              ],
+              [
+                1653,
+                200
+              ],
+              [
+                1653,
+                195
+              ],
+              [
+                1650,
+                192
+              ],
+              [
+                1650,
+                190
+              ],
+              [
+                1648,
+                189
+              ],
+              [
+                1646,
+                189
+              ],
+              [
+                1643,
+                186
+              ],
+              [
+                1635,
+                186
+              ],
+              [
+                1634,
+                184
+              ]
+            ]
+          }
+        ]
+      },
+      "outer_box": {
+        "num": 1,
+        "cls": [
+          1
+        ],
+        "names": [
+          "outer_box"
+        ],
+        "conf": 0.9990946511284866,
+        "shapes": [
+          {
+            "class_num": 1,
+            "label": "outer_box",
+            "probability": 0.9990946511284866,
+            "points": [
+              [
+                1539,
+                134
+              ],
+              [
+                1538,
+                136
+              ],
+              [
+                1426,
+                136
+              ],
+              [
+                1424,
+                138
+              ],
+              [
+                1336,
+                138
+              ],
+              [
+                1334,
+                139
+              ],
+              [
+                1256,
+                139
+              ],
+              [
+                1254,
+                141
+              ],
+              [
+                1147,
+                141
+              ],
+              [
+                1146,
+                142
+              ],
+              [
+                981,
+                142
+              ],
+              [
+                979,
+                144
+              ],
+              [
+                854,
+                144
+              ],
+              [
+                853,
+                146
+              ],
+              [
+                643,
+                146
+              ],
+              [
+                642,
+                147
+              ],
+              [
+                517,
+                147
+              ],
+              [
+                515,
+                149
+              ],
+              [
+                429,
+                149
+              ],
+              [
+                427,
+                150
+              ],
+              [
+                422,
+                150
+              ],
+              [
+                421,
+                152
+              ],
+              [
+                416,
+                152
+              ],
+              [
+                414,
+                154
+              ],
+              [
+                413,
+                154
+              ],
+              [
+                411,
+                155
+              ],
+              [
+                410,
+                155
+              ],
+              [
+                408,
+                157
+              ],
+              [
+                406,
+                157
+              ],
+              [
+                405,
+                158
+              ],
+              [
+                403,
+                158
+              ],
+              [
+                398,
+                163
+              ],
+              [
+                397,
+                163
+              ],
+              [
+                390,
+                170
+              ],
+              [
+                390,
+                171
+              ],
+              [
+                386,
+                176
+              ],
+              [
+                386,
+                178
+              ],
+              [
+                382,
+                181
+              ],
+              [
+                382,
+                182
+              ],
+              [
+                381,
+                184
+              ],
+              [
+                381,
+                186
+              ],
+              [
+                379,
+                187
+              ],
+              [
+                379,
+                192
+              ],
+              [
+                378,
+                194
+              ],
+              [
+                378,
+                205
+              ],
+              [
+                376,
+                206
+              ],
+              [
+                376,
+                360
+              ],
+              [
+                378,
+                362
+              ],
+              [
+                378,
+                710
+              ],
+              [
+                379,
+                712
+              ],
+              [
+                379,
+                917
+              ],
+              [
+                381,
+                918
+              ],
+              [
+                381,
+                1093
+              ],
+              [
+                382,
+                1094
+              ],
+              [
+                382,
+                1250
+              ],
+              [
+                384,
+                1251
+              ],
+              [
+                384,
+                1365
+              ],
+              [
+                386,
+                1366
+              ],
+              [
+                386,
+                1494
+              ],
+              [
+                387,
+                1496
+              ],
+              [
+                387,
+                1634
+              ],
+              [
+                389,
+                1635
+              ],
+              [
+                389,
+                1744
+              ],
+              [
+                390,
+                1746
+              ],
+              [
+                390,
+                1867
+              ],
+              [
+                392,
+                1869
+              ],
+              [
+                392,
+                1941
+              ],
+              [
+                394,
+                1942
+              ],
+              [
+                394,
+                1954
+              ],
+              [
+                395,
+                1955
+              ],
+              [
+                395,
+                1960
+              ],
+              [
+                397,
+                1962
+              ],
+              [
+                397,
+                1965
+              ],
+              [
+                398,
+                1966
+              ],
+              [
+                398,
+                1968
+              ],
+              [
+                400,
+                1970
+              ],
+              [
+                400,
+                1971
+              ],
+              [
+                402,
+                1973
+              ],
+              [
+                402,
+                1974
+              ],
+              [
+                403,
+                1976
+              ],
+              [
+                403,
+                1978
+              ],
+              [
+                416,
+                1990
+              ],
+              [
+                418,
+                1990
+              ],
+              [
+                419,
+                1992
+              ],
+              [
+                421,
+                1992
+              ],
+              [
+                424,
+                1995
+              ],
+              [
+                426,
+                1995
+              ],
+              [
+                427,
+                1997
+              ],
+              [
+                430,
+                1997
+              ],
+              [
+                432,
+                1998
+              ],
+              [
+                434,
+                1998
+              ],
+              [
+                435,
+                2000
+              ],
+              [
+                438,
+                2000
+              ],
+              [
+                440,
+                2002
+              ],
+              [
+                448,
+                2002
+              ],
+              [
+                450,
+                2003
+              ],
+              [
+                526,
+                2003
+              ],
+              [
+                528,
+                2002
+              ],
+              [
+                622,
+                2002
+              ],
+              [
+                624,
+                2000
+              ],
+              [
+                779,
+                2000
+              ],
+              [
+                781,
+                1998
+              ],
+              [
+                848,
+                1998
+              ],
+              [
+                850,
+                1997
+              ],
+              [
+                939,
+                1997
+              ],
+              [
+                941,
+                1995
+              ],
+              [
+                1070,
+                1995
+              ],
+              [
+                1072,
+                1994
+              ],
+              [
+                1235,
+                1994
+              ],
+              [
+                1237,
+                1992
+              ],
+              [
+                1539,
+                1992
+              ],
+              [
+                1541,
+                1990
+              ],
+              [
+                1618,
+                1990
+              ],
+              [
+                1619,
+                1989
+              ],
+              [
+                1672,
+                1989
+              ],
+              [
+                1674,
+                1987
+              ],
+              [
+                1678,
+                1987
+              ],
+              [
+                1680,
+                1986
+              ],
+              [
+                1683,
+                1986
+              ],
+              [
+                1685,
+                1984
+              ],
+              [
+                1686,
+                1984
+              ],
+              [
+                1688,
+                1982
+              ],
+              [
+                1690,
+                1982
+              ],
+              [
+                1691,
+                1981
+              ],
+              [
+                1693,
+                1981
+              ],
+              [
+                1698,
+                1976
+              ],
+              [
+                1699,
+                1976
+              ],
+              [
+                1712,
+                1963
+              ],
+              [
+                1712,
+                1962
+              ],
+              [
+                1715,
+                1958
+              ],
+              [
+                1715,
+                1957
+              ],
+              [
+                1717,
+                1955
+              ],
+              [
+                1717,
+                1954
+              ],
+              [
+                1718,
+                1952
+              ],
+              [
+                1718,
+                1949
+              ],
+              [
+                1720,
+                1947
+              ],
+              [
+                1720,
+                1942
+              ],
+              [
+                1722,
+                1941
+              ],
+              [
+                1722,
+                1930
+              ],
+              [
+                1723,
+                1928
+              ],
+              [
+                1723,
+                1669
+              ],
+              [
+                1722,
+                1667
+              ],
+              [
+                1722,
+                1406
+              ],
+              [
+                1720,
+                1405
+              ],
+              [
+                1720,
+                1230
+              ],
+              [
+                1718,
+                1229
+              ],
+              [
+                1718,
+                1090
+              ],
+              [
+                1717,
+                1088
+              ],
+              [
+                1717,
+                931
+              ],
+              [
+                1715,
+                930
+              ],
+              [
+                1715,
+                819
+              ],
+              [
+                1714,
+                818
+              ],
+              [
+                1714,
+                690
+              ],
+              [
+                1712,
+                688
+              ],
+              [
+                1712,
+                542
+              ],
+              [
+                1710,
+                541
+              ],
+              [
+                1710,
+                408
+              ],
+              [
+                1709,
+                406
+              ],
+              [
+                1709,
+                318
+              ],
+              [
+                1707,
+                317
+              ],
+              [
+                1707,
+                213
+              ],
+              [
+                1706,
+                211
+              ],
+              [
+                1706,
+                184
+              ],
+              [
+                1704,
+                182
+              ],
+              [
+                1704,
+                179
+              ],
+              [
+                1702,
+                178
+              ],
+              [
+                1702,
+                174
+              ],
+              [
+                1701,
+                173
+              ],
+              [
+                1701,
+                171
+              ],
+              [
+                1699,
+                170
+              ],
+              [
+                1699,
+                168
+              ],
+              [
+                1698,
+                166
+              ],
+              [
+                1698,
+                165
+              ],
+              [
+                1696,
+                163
+              ],
+              [
+                1696,
+                162
+              ],
+              [
+                1680,
+                146
+              ],
+              [
+                1678,
+                146
+              ],
+              [
+                1677,
+                144
+              ],
+              [
+                1675,
+                144
+              ],
+              [
+                1672,
+                141
+              ],
+              [
+                1670,
+                141
+              ],
+              [
+                1669,
+                139
+              ],
+              [
+                1666,
+                139
+              ],
+              [
+                1664,
+                138
+              ],
+              [
+                1661,
+                138
+              ],
+              [
+                1659,
+                136
+              ],
+              [
+                1651,
+                136
+              ],
+              [
+                1650,
+                134
+              ]
+            ]
+          }
+        ]
+      }
+    },
+    "defect_inference": {
+      "defect_count": 5,
+      "total_actual_area": 1.356,
+      "defects": [
+        {
+          "label": "wear",
+          "pixel_area": 127.5,
+          "actual_area": 0.07678197899999999,
+          "width": 0.76074,
+          "height": 0.17178,
+          "contour": [
+            [
+              275,
+              169
+            ],
+            [
+              277,
+              171
+            ],
+            [
+              282,
+              172
+            ],
+            [
+              279,
+              173
+            ],
+            [
+              280,
+              175
+            ],
+            [
+              288,
+              175
+            ],
+            [
+              295,
+              174
+            ],
+            [
+              295,
+              171
+            ],
+            [
+              299,
+              171
+            ],
+            [
+              303,
+              170
+            ],
+            [
+              306,
+              170
+            ],
+            [
+              305,
+              168
+            ],
+            [
+              298,
+              168
+            ],
+            [
+              288,
+              169
+            ],
+            [
+              281,
+              169
+            ],
+            [
+              277,
+              168
+            ]
+          ],
+          "min_rect": [
+            [
+              290.5,
+              171.5
+            ],
+            [
+              31.0,
+              7.0
+            ],
+            -0.0
+          ]
+        },
+        {
+          "label": "wear",
+          "pixel_area": 37.0,
+          "actual_area": 0.0222818292,
+          "width": 0.18624548626899717,
+          "height": 0.1552045777416229,
+          "contour": [
+            [
+              135,
+              210
+            ],
+            [
+              137,
+              213
+            ],
+            [
+              137,
+              216
+            ],
+            [
+              135,
+              218
+            ],
+            [
+              132,
+              218
+            ],
+            [
+              131,
+              215
+            ],
+            [
+              132,
+              211
+            ]
+          ],
+          "min_rect": [
+            [
+              134.09999084472656,
+              214.30001831054688
+            ],
+            [
+              7.589465618133545,
+              6.324554920196533
+            ],
+            71.56504821777344
+          ]
+        },
+        {
+          "label": "wear",
+          "pixel_area": 160.5,
+          "actual_area": 0.0966549618,
+          "width": 0.16266578845024107,
+          "height": 0.9137169589233398,
+          "contour": [
+            [
+              123,
+              227
+            ],
+            [
+              125,
+              232
+            ],
+            [
+              123,
+              235
+            ],
+            [
+              121,
+              235
+            ],
+            [
+              121,
+              239
+            ],
+            [
+              120,
+              241
+            ],
+            [
+              118,
+              244
+            ],
+            [
+              116,
+              247
+            ],
+            [
+              115,
+              247
+            ],
+            [
+              114,
+              251
+            ],
+            [
+              114,
+              256
+            ],
+            [
+              111,
+              258
+            ],
+            [
+              110,
+              262
+            ],
+            [
+              108,
+              260
+            ],
+            [
+              107,
+              257
+            ],
+            [
+              109,
+              251
+            ],
+            [
+              112,
+              245
+            ],
+            [
+              115,
+              240
+            ],
+            [
+              118,
+              234
+            ],
+            [
+              121,
+              229
+            ]
+          ],
+          "min_rect": [
+            [
+              115.77690124511719,
+              244.1685791015625
+            ],
+            [
+              6.628597736358643,
+              37.23377990722656
+            ],
+            24.623563766479492
+          ]
+        },
+        {
+          "label": "wear",
+          "pixel_area": 199.0,
+          "actual_area": 0.11984010839999999,
+          "width": 0.22834019067764283,
+          "height": 0.8093390180969238,
+          "contour": [
+            [
+              425,
+              167
+            ],
+            [
+              426,
+              169
+            ],
+            [
+              428,
+              169
+            ],
+            [
+              428,
+              173
+            ],
+            [
+              430,
+              175
+            ],
+            [
+              434,
+              176
+            ],
+            [
+              436,
+              173
+            ],
+            [
+              437,
+              171
+            ],
+            [
+              440,
+              172
+            ],
+            [
+              447,
+              172
+            ],
+            [
+              448,
+              174
+            ],
+            [
+              449,
+              175
+            ],
+            [
+              453,
+              174
+            ],
+            [
+              457,
+              172
+            ],
+            [
+              458,
+              169
+            ],
+            [
+              458,
+              167
+            ],
+            [
+              454,
+              166
+            ],
+            [
+              445,
+              167
+            ],
+            [
+              434,
+              167
+            ],
+            [
+              428,
+              167
+            ]
+          ],
+          "min_rect": [
+            [
+              441.6407165527344,
+              171.08135986328125
+            ],
+            [
+              9.304816246032715,
+              32.98040008544922
+            ],
+            88.02507019042969
+          ]
+        },
+        {
+          "label": "wear",
+          "pixel_area": 308.0,
+          "actual_area": 0.18548117279999998,
+          "width": 0.24939382078170777,
+          "height": 1.1079625957489014,
+          "contour": [
+            [
+              501,
+              173
+            ],
+            [
+              505,
+              171
+            ],
+            [
+              510,
+              169
+            ],
+            [
+              511,
+              167
+            ],
+            [
+              510,
+              166
+            ],
+            [
+              504,
+              165
+            ],
+            [
+              494,
+              166
+            ],
+            [
+              487,
+              166
+            ],
+            [
+              479,
+              166
+            ],
+            [
+              471,
+              166
+            ],
+            [
+              468,
+              166
+            ],
+            [
+              467,
+              168
+            ],
+            [
+              466,
+              170
+            ],
+            [
+              466,
+              173
+            ],
+            [
+              468,
+              173
+            ],
+            [
+              470,
+              173
+            ],
+            [
+              471,
+              174
+            ],
+            [
+              474,
+              176
+            ],
+            [
+              479,
+              175
+            ],
+            [
+              482,
+              174
+            ],
+            [
+              486,
+              173
+            ],
+            [
+              488,
+              173
+            ],
+            [
+              491,
+              172
+            ],
+            [
+              493,
+              171
+            ],
+            [
+              494,
+              170
+            ],
+            [
+              496,
+              171
+            ],
+            [
+              496,
+              173
+            ],
+            [
+              497,
+              173
+            ],
+            [
+              499,
+              173
+            ]
+          ],
+          "min_rect": [
+            [
+              488.51434326171875,
+              170.51348876953125
+            ],
+            [
+              10.162747383117676,
+              45.14925003051758
+            ],
+            88.40885925292969
+          ]
+        }
+      ]
+    }
+  }
+}