aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 git忽略文件配置, triton地址修改

Browse Source
This commit is contained in:
zhouchengrong
2024-03-19 17:04:41 +08:00
parent 04d1194dc6
commit c536e1ad36
1 changed files with 61 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
app/service/attribute_recognition/const.py
View File

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