ML-Motivators commited on
Commit
1b1d6da
·
verified ·
1 Parent(s): 14a8e56

Delete densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml

Browse files
densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml DELETED
@@ -1,133 +0,0 @@
1
- _BASE_: "Base-DensePose-RCNN-FPN.yaml"
2
- MODEL:
3
- WEIGHTS: "https://dl.fbaipublicfiles.com/densepose/cse/densepose_rcnn_R_50_FPN_soft_s1x/250533982/model_final_2c4512.pkl"
4
- RESNETS:
5
- DEPTH: 50
6
- ROI_HEADS:
7
- NUM_CLASSES: 1
8
- ROI_DENSEPOSE_HEAD:
9
- NAME: "DensePoseV1ConvXHead"
10
- COARSE_SEGM_TRAINED_BY_MASKS: True
11
- CSE:
12
- EMBED_LOSS_NAME: "SoftEmbeddingLoss"
13
- EMBEDDING_DIST_GAUSS_SIGMA: 0.1
14
- GEODESIC_DIST_GAUSS_SIGMA: 0.1
15
- EMBEDDERS:
16
- "cat_5001":
17
- TYPE: vertex_feature
18
- NUM_VERTICES: 5001
19
- FEATURE_DIM: 256
20
- FEATURES_TRAINABLE: False
21
- IS_TRAINABLE: True
22
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_cat_5001_256.pkl"
23
- "dog_5002":
24
- TYPE: vertex_feature
25
- NUM_VERTICES: 5002
26
- FEATURE_DIM: 256
27
- FEATURES_TRAINABLE: False
28
- IS_TRAINABLE: True
29
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_dog_5002_256.pkl"
30
- "sheep_5004":
31
- TYPE: vertex_feature
32
- NUM_VERTICES: 5004
33
- FEATURE_DIM: 256
34
- FEATURES_TRAINABLE: False
35
- IS_TRAINABLE: True
36
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_sheep_5004_256.pkl"
37
- "horse_5004":
38
- TYPE: vertex_feature
39
- NUM_VERTICES: 5004
40
- FEATURE_DIM: 256
41
- FEATURES_TRAINABLE: False
42
- IS_TRAINABLE: True
43
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_horse_5004_256.pkl"
44
- "zebra_5002":
45
- TYPE: vertex_feature
46
- NUM_VERTICES: 5002
47
- FEATURE_DIM: 256
48
- FEATURES_TRAINABLE: False
49
- IS_TRAINABLE: True
50
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_zebra_5002_256.pkl"
51
- "giraffe_5002":
52
- TYPE: vertex_feature
53
- NUM_VERTICES: 5002
54
- FEATURE_DIM: 256
55
- FEATURES_TRAINABLE: False
56
- IS_TRAINABLE: True
57
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_giraffe_5002_256.pkl"
58
- "elephant_5002":
59
- TYPE: vertex_feature
60
- NUM_VERTICES: 5002
61
- FEATURE_DIM: 256
62
- FEATURES_TRAINABLE: False
63
- IS_TRAINABLE: True
64
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_elephant_5002_256.pkl"
65
- "cow_5002":
66
- TYPE: vertex_feature
67
- NUM_VERTICES: 5002
68
- FEATURE_DIM: 256
69
- FEATURES_TRAINABLE: False
70
- IS_TRAINABLE: True
71
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_cow_5002_256.pkl"
72
- "bear_4936":
73
- TYPE: vertex_feature
74
- NUM_VERTICES: 4936
75
- FEATURE_DIM: 256
76
- FEATURES_TRAINABLE: False
77
- IS_TRAINABLE: True
78
- INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_bear_4936_256.pkl"
79
- DATASETS:
80
- TRAIN:
81
- - "densepose_lvis_v1_ds1_train_v1"
82
- TEST:
83
- - "densepose_lvis_v1_ds1_val_v1"
84
- WHITELISTED_CATEGORIES:
85
- "densepose_lvis_v1_ds1_train_v1":
86
- - 943 # sheep
87
- - 1202 # zebra
88
- - 569 # horse
89
- - 496 # giraffe
90
- - 422 # elephant
91
- - 80 # cow
92
- - 76 # bear
93
- - 225 # cat
94
- - 378 # dog
95
- "densepose_lvis_v1_ds1_val_v1":
96
- - 943 # sheep
97
- - 1202 # zebra
98
- - 569 # horse
99
- - 496 # giraffe
100
- - 422 # elephant
101
- - 80 # cow
102
- - 76 # bear
103
- - 225 # cat
104
- - 378 # dog
105
- CATEGORY_MAPS:
106
- "densepose_lvis_v1_ds1_train_v1":
107
- "1202": 943 # zebra -> sheep
108
- "569": 943 # horse -> sheep
109
- "496": 943 # giraffe -> sheep
110
- "422": 943 # elephant -> sheep
111
- "80": 943 # cow -> sheep
112
- "76": 943 # bear -> sheep
113
- "225": 943 # cat -> sheep
114
- "378": 943 # dog -> sheep
115
- "densepose_lvis_v1_ds1_val_v1":
116
- "1202": 943 # zebra -> sheep
117
- "569": 943 # horse -> sheep
118
- "496": 943 # giraffe -> sheep
119
- "422": 943 # elephant -> sheep
120
- "80": 943 # cow -> sheep
121
- "76": 943 # bear -> sheep
122
- "225": 943 # cat -> sheep
123
- "378": 943 # dog -> sheep
124
- CLASS_TO_MESH_NAME_MAPPING:
125
- # Note: different classes are mapped to a single class
126
- # mesh is chosen based on GT data, so this is just some
127
- # value which has no particular meaning
128
- "0": "sheep_5004"
129
- SOLVER:
130
- MAX_ITER: 4000
131
- STEPS: (3000, 3500)
132
- DENSEPOSE_EVALUATION:
133
- EVALUATE_MESH_ALIGNMENT: True