属性检测接口，响应字段规范
This commit is contained in:
zhouchengrong
139
app/service/attribute_recognition/const_debug.py
Normal file
139
app/service/attribute_recognition/const_debug.py
Normal file
@@ -0,0 +1,139 @@
|
||||
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'
|
||||
Reference in New Issue
Block a user