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',