140 lines
8.4 KiB
Python
140 lines
8.4 KiB
Python
import torch
|
|
|
|
device = torch.device('cuda')
|
|
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',
|
|
'top_Sleeve_length',
|
|
'top_Sleeve_shape',
|
|
'top_Sleeve_shoulder',
|
|
'top_Neckline',
|
|
'top_print',
|
|
'top_material',
|
|
'top_Softness',
|
|
'top_Design',
|
|
'top_optype',
|
|
'top_Silhouette',
|
|
'top_Collar'
|
|
]
|
|
|
|
bottom_discription_list = [
|
|
'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',
|
|
'bottom_length',
|
|
'bottom_print',
|
|
'bottom_material',
|
|
'bottom_Softness_B',
|
|
'bottom_Silhouette_B',
|
|
'bottom_OPType_B',
|
|
'bottom_design']
|
|
|
|
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',
|
|
'outwear_outer_sleeve_length',
|
|
'outwear_outer_sleeve_shape',
|
|
'outwear_outer_sleeve_shoulder',
|
|
'outwear_outer_collar',
|
|
'outwear_print',
|
|
'outwear_material',
|
|
'outwear_outer_softness',
|
|
'outwear_outer_design',
|
|
# 'outwear_11_outer_opening',
|
|
'outwear_outer_optype',
|
|
'outwear_outer_silhouette']
|
|
|
|
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',
|
|
'jumpsuit_sleeve_shape',
|
|
'jumpsuit_sleeve_shoulder',
|
|
'jumpsuit_neckline',
|
|
'jumpsuit_collar',
|
|
'jumpsuit_print',
|
|
'jumpsuit_material',
|
|
'jumpsuit_softness',
|
|
'jumpsuit_design',
|
|
'jumpsuit_optype',
|
|
'jumpsuit_subtype']
|
|
|
|
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',
|
|
'dress_sleeve_shape',
|
|
'dress_sleeve_shoulder',
|
|
'dress_neckline',
|
|
'dress_print',
|
|
'dress_collar',
|
|
'dress_material',
|
|
'dress_design',
|
|
'dress_softness',
|
|
'dress_silohouette12',
|
|
# 'dress_'
|
|
'dress_type'
|
|
]
|
|
|
|
category_discription = 'app/service/attribute_recognition/discriptor/category/category_dis.csv'
|
|
category_model = 'category'
|