关于保存特征的一些代码
This commit is contained in:
@@ -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")
|
||||
|
||||
16
app/api/api_simiar_match.py
Normal file
16
app/api/api_simiar_match.py
Normal 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
|
||||
Reference in New Issue
Block a user