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

fix 类别映射表修改

Browse Source
This commit is contained in:
zhouchengrong
2024-08-29 14:58:30 +08:00
parent 94a5088f37
commit f4c504e8b0
1 changed files with 5 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app/service/attribute_recognition/service.py
View File

@@ -1,17 +1,13 @@
import datetime
from pprint import pprint from pprint import pprint
import mmcv
import cv2 import cv2
import mmcv
import numpy as np import numpy as np
import requests import pandas as pd
import torch import torch
import tritonclient.http as httpclient import tritonclient.http as httpclient
from PIL import Image
from skimage import transform
from torchvision import transforms
import pandas as pd
from minio import Minio from minio import Minio
from skimage import transform
from app.core.config import MINIO_IP, MINIO_PORT, MINIO_SECURE, MINIO_ACCESS, MINIO_SECRET, ATT_TRITON_IP, ATT_TRITON_PORT from app.core.config import MINIO_IP, MINIO_PORT, MINIO_SECURE, MINIO_ACCESS, MINIO_SECRET, ATT_TRITON_IP, ATT_TRITON_PORT
from app.service.attribute_recognition.config.category_mapping import category_mapping from app.service.attribute_recognition.config.category_mapping import category_mapping
@@ -107,7 +103,7 @@ class AttributeRecognition:
print(category) print(category)
if category == 'tops': if category == 'tops':
attr_dict = {'Item': category} attr_dict = {'Item': "top"}
for i in range(len(args.top_discription_list)): for i in range(len(args.top_discription_list)):
attr_description = args.top_discription_list[i] attr_description = args.top_discription_list[i]
attr_model_path = args.top_model_list[i] attr_model_path = args.top_model_list[i]
@@ -115,10 +111,7 @@ class AttributeRecognition:
attr_dict = Merge(attr_dict, present_dict) attr_dict = Merge(attr_dict, present_dict)
elif category == 'bottoms': elif category == 'bottoms':
attr_dict = {} attr_dict = {'Item': "bottom", 'Type': ['Pants']}
category = 'bottom'
attr_dict['Item'] = category
attr_dict['Type'] = ['Pants']
for i in range(len(args.bottom_discription_list)): for i in range(len(args.bottom_discription_list)):
attr_description = args.bottom_discription_list[i] attr_description = args.bottom_discription_list[i]
attr_model_path = args.bottom_model_list[i] attr_model_path = args.bottom_model_list[i]