From f4c504e8b0fc41b102d256d607c48ff0c38cfe92 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 29 Aug 2024 14:58:30 +0800 Subject: [PATCH] =?UTF-8?q?fix=20=E7=B1=BB=E5=88=AB=E6=98=A0=E5=B0=84?= =?UTF-8?q?=E8=A1=A8=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/attribute_recognition/service.py | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) 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]