diff --git a/app/service/attribute_recognition/service.py b/app/service/attribute_recognition/service.py index 16a5bef..2044086 100644 --- a/app/service/attribute_recognition/service.py +++ b/app/service/attribute_recognition/service.py @@ -1,17 +1,13 @@ -import datetime from pprint import pprint -import mmcv import cv2 +import mmcv import numpy as np -import requests +import pandas as pd import torch 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 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.service.attribute_recognition.config.category_mapping import category_mapping @@ -107,7 +103,7 @@ class AttributeRecognition: print(category) if category == 'tops': - attr_dict = {'Item': category} + attr_dict = {'Item': "top"} for i in range(len(args.top_discription_list)): attr_description = args.top_discription_list[i] attr_model_path = args.top_model_list[i] @@ -115,10 +111,7 @@ class AttributeRecognition: attr_dict = Merge(attr_dict, present_dict) elif category == 'bottoms': - attr_dict = {} - category = 'bottom' - attr_dict['Item'] = category - attr_dict['Type'] = ['Pants'] + attr_dict = {'Item': "bottom", 'Type': ['Pants']} for i in range(len(args.bottom_discription_list)): attr_description = args.bottom_discription_list[i] attr_model_path = args.bottom_model_list[i]