fix 类别映射表修改

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]