aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新一些配置与配置相关的代码

Browse Source
This commit is contained in:
zhouchengrong
2024-03-18 17:57:01 +08:00
parent 8383234ba1
commit f84b8aaba1
6 changed files with 71 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/service/outfit_matcher/dataset.py
View File

@@ -4,6 +4,8 @@ import itertools
import difflib
from copy import deepcopy
from app.core.config import FASHION_CATEGORIES, FASHION_CATEGORIES_MAPPING
# Helper function to calculate the edit distance similarity between two file names
def sim_score(a, b):
@@ -11,12 +13,8 @@ def sim_score(a, b):
class FashionDataset(object):
fashion_categories = json.load(open(r"app/service/outfit_matcher/config/fashion_categories.json", "r"))
fashion_categories_mapping = json.load(open(r"app/service/outfit_matcher/config/fashion_category_mapping.json", "r"))
# service debug
# fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.json", "r"))
# fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r"))
fashion_categories = json.load(open(FASHION_CATEGORIES, "r"))
fashion_categories_mapping = json.load(open(FASHION_CATEGORIES_MAPPING, "r"))
def __init__(self, item_metadata):
self.item_metadat = item_metadata

2
app/service/outfit_matcher/outfit_evaluator.py
View File

@@ -15,7 +15,7 @@ from app.service.utils.decorator import RunTime
class OutfitMatcher(object):
def __init__(self):
self.tritonclient = httpclient.InferenceServerClient(url="10.1.1.240:10010")
self.tritonclient = httpclient.InferenceServerClient(url=f"{OM_TRITON_IP}:{OM_TRITON_PORT}")
self.minio_client = Minio(
f"{MINIO_IP}:{MINIO_PORT}",
access_key=MINIO_ACCESS,