Picture using minio service

app/api/api_attribute.py

@@ -0,0 +1,18 @@
+import logging
+
+from fastapi import APIRouter
+
+from app.schemas.attribute import AttributeModel
+from app.service.attribute_recognition import const
+from app.service.attribute_recognition.service import AttributeRecognition
+
+logger = logging.getLogger()
+router = APIRouter()
+
+
+@router.post("")
+def attribute(request_data: AttributeModel):
+    service = AttributeRecognition()
+    response = service.attribute(const, request_data)
+    logger.info("test")
+    return {"code": 200, "message": "ok", "data": response}

app/api/api_outfit_matcher.py

@@ -1,15 +1,44 @@
 import logging
+import time
 
 from fastapi import APIRouter
-
-from app.service.outfit_matcher.service import OutfitMatcherHon
+from app.schemas.outfit_matcher import OutfitMatcher
+from app.service.outfit_matcher.dataset import FashionDataset
+from app.service.outfit_matcher.outfit_evaluator import OutfitMaterTypeAware
+from app.service.utils.decorator import RunTime
 
 logger = logging.getLogger()
 router = APIRouter()
 
 
-@router.post("")
-def outfit_matcher_hon():
-    service = OutfitMatcherHon()
-    logger.info("test")
-    return {"message": "ok"}
+@RunTime
+@router.post("outfit_matcher")
+def outfit_matcher(request_item: OutfitMatcher):
+    request_item = dict(request_item)
+    for i in range(len(request_item['query'])):
+        request_item['query'][i] = dict(request_item['query'][i])
+    for i in range(len(request_item['database'])):
+        request_item['database'][i] = dict(request_item['database'][i])
+
+    try:
+        fashion_dataset = FashionDataset(request_item['database'])
+        service = OutfitMaterTypeAware()
+        result = []
+        start_time = time.time()
+        for item in request_item['query']:
+            outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])
+            scores = service.get_result(outfits)
+            if request_item['is_best']:
+                best_outfits, best_scores = service.visualize(outfits, scores, request_item["topk"], best=True,
+                                                              # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_best_{param['topk']}.png")
+                                                              )
+                result.append({"best_outfits": best_outfits, "best_scores": best_scores})
+            else:
+                bad_outfits, bad_scores = service.visualize(outfits, scores, request_item["topk"], best=False,
+                                                            # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png")
+                                                            )
+                result.append({"bad_outfits": bad_outfits, "bad_scores": bad_scores})
+        logger.info(f"run time is : {time.time() - start_time}")
+        return {"message": "ok", "data": result}
+    except Exception as e:
+        return {"message": f"{e}", "data": e}

app/api/api_route.py

@@ -2,8 +2,10 @@ from fastapi import APIRouter
 
 from app.api import api_test
 from app.api import api_outfit_matcher
+from app.api import api_attribute
 
 router = APIRouter()
 
 router.include_router(api_test.router, tags=["test"], prefix="/test")
-router.include_router(api_outfit_matcher.router, tags=["outfit_matcher"], prefix="/outfit_matcher")
+router.include_router(api_outfit_matcher.router, tags=["outfit_matcher"], prefix="/api/outfit_matcher")
+router.include_router(api_attribute.router, tags=["attribute"], prefix="/api/attribute")