关于保存特征的一些代码

This commit is contained in:
zchen
2024-03-26 18:00:35 +08:00
parent 7dce70027a
commit 88c815dd9d
16 changed files with 378 additions and 10 deletions

View File

@@ -27,7 +27,9 @@ def outfit_matcher(request_item: OutfitMatcher):
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)
scores, features = service.get_result(outfits)
# save features in databases
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")

View File

@@ -0,0 +1,16 @@
import logging
import time
from fastapi import APIRouter
from app.schemas.outfit_matcher import SimilarMatchMItem
from app.service.utils.decorator import RunTime
logger = logging.getLogger()
router = APIRouter()
@RunTime
@router.post("similar_match")
def similar_match(request_item: SimilarMatchMItem):
pass