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