From c536e1ad3655ee539f6fb0618c45f9f52879226b Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Tue, 19 Mar 2024 17:04:41 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20git=E5=BF=BD=E7=95=A5?= =?UTF-8?q?=E6=96=87=E4=BB=B6=E9=85=8D=E7=BD=AE,=20triton=E5=9C=B0?= =?UTF-8?q?=E5=9D=80=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/const.py | 122 ++++++++++----------- 1 file changed, 61 insertions(+), 61 deletions(-) diff --git a/app/service/attribute_recognition/const.py b/app/service/attribute_recognition/const.py index fab4fd0..a8be474 100644 --- a/app/service/attribute_recognition/const.py +++ b/app/service/attribute_recognition/const.py @@ -1,20 +1,20 @@ import torch device = torch.device('cuda') -top_discription_list = ['service/attribute_recognition/discriptor/top/1_top_length.csv', - 'service/attribute_recognition/discriptor/top/2_top_type.csv', - 'service/attribute_recognition/discriptor/top/3_top_Sleeve_length.csv', - 'service/attribute_recognition/discriptor/top/4_top_Sleeve_shape.csv', - 'service/attribute_recognition/discriptor/top/5_top_Sleeve_shoulder.csv', - 'service/attribute_recognition/discriptor/top/6_top_Neckline.csv', - 'service/attribute_recognition/discriptor/top/7_outer_Print.csv', - 'service/attribute_recognition/discriptor/top/8_outer_Material.csv', - # 'service/attribute_recognition/discriptor/top/9_top_Material.csv', - 'service/attribute_recognition/discriptor/top/9_top_Softness.csv', - 'service/attribute_recognition/discriptor/top/10_top_Design.csv', - 'service/attribute_recognition/discriptor/top/11_top_OPType.csv', - 'service/attribute_recognition/discriptor/top/12_top_Silhouette.csv', - 'service/attribute_recognition/discriptor/top/7_top_Collar.csv'] +top_discription_list = ['app/service/attribute_recognition/discriptor/top/1_top_length.csv', + 'app/service/attribute_recognition/discriptor/top/2_top_type.csv', + 'app/service/attribute_recognition/discriptor/top/3_top_Sleeve_length.csv', + 'app/service/attribute_recognition/discriptor/top/4_top_Sleeve_shape.csv', + 'app/service/attribute_recognition/discriptor/top/5_top_Sleeve_shoulder.csv', + 'app/service/attribute_recognition/discriptor/top/6_top_Neckline.csv', + 'app/service/attribute_recognition/discriptor/top/7_outer_Print.csv', + 'app/service/attribute_recognition/discriptor/top/8_outer_Material.csv', + # 'app/service/attribute_recognition/discriptor/top/9_top_Material.csv', + 'app/service/attribute_recognition/discriptor/top/9_top_Softness.csv', + 'app/service/attribute_recognition/discriptor/top/10_top_Design.csv', + 'app/service/attribute_recognition/discriptor/top/11_top_OPType.csv', + 'app/service/attribute_recognition/discriptor/top/12_top_Silhouette.csv', + 'app/service/attribute_recognition/discriptor/top/7_top_Collar.csv'] top_model_list = ['top_length', 'top_type', @@ -32,15 +32,15 @@ top_model_list = ['top_length', ] bottom_discription_list = [ - 'service/attribute_recognition/discriptor/bottom/2_bottom_subtype.csv', - # 'service/attribute_recognition/discriptor/bottom/3_bottom_structure.csv', - 'service/attribute_recognition/discriptor/bottom/3_bottom_length.csv', - 'service/attribute_recognition/discriptor/bottom/7_outer_Print.csv', - 'service/attribute_recognition/discriptor/bottom/8_outer_Material.csv', - 'service/attribute_recognition/discriptor/bottom/5_bottom_Softness.csv', - 'service/attribute_recognition/discriptor/bottom/8_bottom_Silhouette.csv', - 'service/attribute_recognition/discriptor/bottom/7_bottom_OPType.csv', - 'service/attribute_recognition/discriptor/bottom/6_bottom_Design.csv'] + 'app/service/attribute_recognition/discriptor/bottom/2_bottom_subtype.csv', + # 'app/service/attribute_recognition/discriptor/bottom/3_bottom_structure.csv', + 'app/service/attribute_recognition/discriptor/bottom/3_bottom_length.csv', + 'app/service/attribute_recognition/discriptor/bottom/7_outer_Print.csv', + 'app/service/attribute_recognition/discriptor/bottom/8_outer_Material.csv', + 'app/service/attribute_recognition/discriptor/bottom/5_bottom_Softness.csv', + 'app/service/attribute_recognition/discriptor/bottom/8_bottom_Silhouette.csv', + 'app/service/attribute_recognition/discriptor/bottom/7_bottom_OPType.csv', + 'app/service/attribute_recognition/discriptor/bottom/6_bottom_Design.csv'] bottom_model_list = [ 'bottom_sub-Type', @@ -52,19 +52,19 @@ bottom_model_list = [ 'bottom_OPType_B', 'bottom_Design_B'] -outwear_discription_list = ['service/attribute_recognition/discriptor/outwear/1_outer_length.csv', - # 'service/attribute_recognition/discriptor/outwear/2_outer_type.csv', - 'service/attribute_recognition/discriptor/outwear/3_outer_sleeve_length.csv', - 'service/attribute_recognition/discriptor/outwear/4_outer_sleeve_shape.csv', - 'service/attribute_recognition/discriptor/outwear/5_outer_sleeve_shoulder.csv', - 'service/attribute_recognition/discriptor/outwear/6_outer_Collar.csv', - 'service/attribute_recognition/discriptor/outwear/7_outer_Print.csv', - 'service/attribute_recognition/discriptor/outwear/8_outer_Material.csv', - 'service/attribute_recognition/discriptor/outwear/9_outer_Softness.csv', - 'service/attribute_recognition/discriptor/outwear/10_outer_Design.csv', - # 'service/attribute_recognition/discriptor/outwear/11_outer_opening.csv', - 'service/attribute_recognition/discriptor/outwear/12_outer_OPType.csv', - 'service/attribute_recognition/discriptor/outwear/13_outer_Silhouette.csv', ] +outwear_discription_list = ['app/service/attribute_recognition/discriptor/outwear/1_outer_length.csv', + # 'app/service/attribute_recognition/discriptor/outwear/2_outer_type.csv', + 'app/service/attribute_recognition/discriptor/outwear/3_outer_sleeve_length.csv', + 'app/service/attribute_recognition/discriptor/outwear/4_outer_sleeve_shape.csv', + 'app/service/attribute_recognition/discriptor/outwear/5_outer_sleeve_shoulder.csv', + 'app/service/attribute_recognition/discriptor/outwear/6_outer_Collar.csv', + 'app/service/attribute_recognition/discriptor/outwear/7_outer_Print.csv', + 'app/service/attribute_recognition/discriptor/outwear/8_outer_Material.csv', + 'app/service/attribute_recognition/discriptor/outwear/9_outer_Softness.csv', + 'app/service/attribute_recognition/discriptor/outwear/10_outer_Design.csv', + # 'app/service/attribute_recognition/discriptor/outwear/11_outer_opening.csv', + 'app/service/attribute_recognition/discriptor/outwear/12_outer_OPType.csv', + 'app/service/attribute_recognition/discriptor/outwear/13_outer_Silhouette.csv', ] outwear_model_list = ['outwear_outer_length', # 'outwear_2_outer_type', @@ -80,18 +80,18 @@ outwear_model_list = ['outwear_outer_length', 'outwear_outer_optype', 'outwear_outer_silhouette'] -jumpsuit_discription_list = ['service/attribute_recognition/discriptor/jumpsuit/1_jumsuit_length.csv', - 'service/attribute_recognition/discriptor/jumpsuit/2_jumpsuit_Sleeve_length.csv', - 'service/attribute_recognition/discriptor/jumpsuit/3_jumpsuit_Sleeve_shape.csv', - 'service/attribute_recognition/discriptor/jumpsuit/4_jumpsuit_Sleeve_shoulder.csv', - 'service/attribute_recognition/discriptor/jumpsuit/5_jumpsuit_Neckline.csv', - 'service/attribute_recognition/discriptor/jumpsuit/6_jumpsuit_Collar.csv', - 'service/attribute_recognition/discriptor/jumpsuit/7_jumpsuit_Print.csv', - 'service/attribute_recognition/discriptor/jumpsuit/8_jumpsuit_Material.csv', - 'service/attribute_recognition/discriptor/jumpsuit/9_jumpsuit_Softness.csv', - 'service/attribute_recognition/discriptor/jumpsuit/10_jumsuit_design.csv', - 'service/attribute_recognition/discriptor/jumpsuit/11_jumpsuit_OPType.csv', - 'service/attribute_recognition/discriptor/jumpsuit/12_jumpsuit_subtype.csv'] +jumpsuit_discription_list = ['app/service/attribute_recognition/discriptor/jumpsuit/1_jumsuit_length.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/2_jumpsuit_Sleeve_length.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/3_jumpsuit_Sleeve_shape.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/4_jumpsuit_Sleeve_shoulder.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/5_jumpsuit_Neckline.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/6_jumpsuit_Collar.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/7_jumpsuit_Print.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/8_jumpsuit_Material.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/9_jumpsuit_Softness.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/10_jumsuit_design.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/11_jumpsuit_OPType.csv', + 'app/service/attribute_recognition/discriptor/jumpsuit/12_jumpsuit_subtype.csv'] jumpsuit_model_list = ['jumpsuit_length', 'jumpsuit_sleeve_length', @@ -106,19 +106,19 @@ jumpsuit_model_list = ['jumpsuit_length', 'jumpsuit_optype', 'jumpsuit_subtype'] -dress_discription_list = ['service/attribute_recognition/discriptor/dress/1_dress_length.csv', - 'service/attribute_recognition/discriptor/dress/3_top_Sleeve_length.csv', - 'service/attribute_recognition/discriptor/dress/4_top_Sleeve_shape.csv', - 'service/attribute_recognition/discriptor/dress/5_top_Sleeve_shoulder.csv', - 'service/attribute_recognition/discriptor/dress/ori5_dress_Neckline.csv', - 'service/attribute_recognition/discriptor/dress/7_outer_Print.csv', - 'service/attribute_recognition/discriptor/dress/7_top_Collar.csv', - 'service/attribute_recognition/discriptor/dress/8_outer_Material.csv', - 'service/attribute_recognition/discriptor/dress/9_dress_Design.csv', - 'service/attribute_recognition/discriptor/dress/9_top_Softness.csv', - 'service/attribute_recognition/discriptor/dress/11_dress_Silhouette.csv', - # 'service/attribute_recognition/discriptor/dress/11_top_OPType.csv', - 'service/attribute_recognition/discriptor/dress/12_dress_type.csv'] +dress_discription_list = ['app/service/attribute_recognition/discriptor/dress/1_dress_length.csv', + 'app/service/attribute_recognition/discriptor/dress/3_top_Sleeve_length.csv', + 'app/service/attribute_recognition/discriptor/dress/4_top_Sleeve_shape.csv', + 'app/service/attribute_recognition/discriptor/dress/5_top_Sleeve_shoulder.csv', + 'app/service/attribute_recognition/discriptor/dress/ori5_dress_Neckline.csv', + 'app/service/attribute_recognition/discriptor/dress/7_outer_Print.csv', + 'app/service/attribute_recognition/discriptor/dress/7_top_Collar.csv', + 'app/service/attribute_recognition/discriptor/dress/8_outer_Material.csv', + 'app/service/attribute_recognition/discriptor/dress/9_dress_Design.csv', + 'app/service/attribute_recognition/discriptor/dress/9_top_Softness.csv', + 'app/service/attribute_recognition/discriptor/dress/11_dress_Silhouette.csv', + # 'app/service/attribute_recognition/discriptor/dress/11_top_OPType.csv', + 'app/service/attribute_recognition/discriptor/dress/12_dress_type.csv'] dress_model_list = ['dress_length', 'dress_sleeve_length',