140 lines
10 KiB
Python
140 lines
10 KiB
Python
import torch
|
|
|
|
device = torch.device('cuda')
|
|
top_discription_list = [r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\1_top_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\2_top_type.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\3_top_Sleeve_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\4_top_Sleeve_shape.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\5_top_Sleeve_shoulder.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\6_top_Neckline.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\7_outer_Print.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\8_outer_Material.csv',
|
|
# r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\9_top_Material.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\9_top_Softness.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\10_top_Design.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\11_top_OPType.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\top\12_top_Silhouette.csv',
|
|
r'E:\workspace\trinity_client_mixi\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 = [
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\2_bottom_subtype.csv',
|
|
# r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\3_bottom_structure.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\3_bottom_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\7_outer_Print.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\8_outer_Material.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\5_bottom_Softness.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\8_bottom_Silhouette.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\bottom\7_bottom_OPType.csv',
|
|
r'E:\workspace\trinity_client_mixi\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 = [r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\1_outer_length.csv',
|
|
# r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\2_outer_type.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\3_outer_sleeve_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\4_outer_sleeve_shape.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\5_outer_sleeve_shoulder.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\6_outer_Collar.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\7_outer_Print.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\8_outer_Material.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\9_outer_Softness.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\10_outer_Design.csv',
|
|
# r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\11_outer_opening.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\outwear\12_outer_OPType.csv',
|
|
r'E:\workspace\trinity_client_mixi\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 = [r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\1_jumsuit_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\2_jumpsuit_Sleeve_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\3_jumpsuit_Sleeve_shape.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\4_jumpsuit_Sleeve_shoulder.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\5_jumpsuit_Neckline.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\6_jumpsuit_Collar.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\7_jumpsuit_Print.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\8_jumpsuit_Material.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\9_jumpsuit_Softness.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\10_jumsuit_design.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\jumpsuit\11_jumpsuit_OPType.csv',
|
|
r'E:\workspace\trinity_client_mixi\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 = [r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\1_dress_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\3_top_Sleeve_length.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\4_top_Sleeve_shape.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\5_top_Sleeve_shoulder.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\ori5_dress_Neckline.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\7_outer_Print.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\7_top_Collar.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\8_outer_Material.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\9_dress_Design.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\9_top_Softness.csv',
|
|
r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\11_dress_Silhouette.csv',
|
|
# r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\dress\11_top_OPType.csv',
|
|
r'E:\workspace\trinity_client_mixi\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 = r'E:\workspace\trinity_client_mixi\app\service\attribute_recognition\discriptor\category\category_dis.csv'
|
|
category_model = 'category'
|