ML-Motivators commited on
Commit
3278c29
·
verified ·
1 Parent(s): 1b1d6da

Delete densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml

Browse files
densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml DELETED
@@ -1,119 +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: 9
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_7466":
17
- TYPE: vertex_feature
18
- NUM_VERTICES: 7466
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_7466_256.pkl"
23
- "dog_7466":
24
- TYPE: vertex_feature
25
- NUM_VERTICES: 7466
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_7466_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_ds2_train_v1"
82
- TEST:
83
- - "densepose_lvis_v1_ds2_val_v1"
84
- WHITELISTED_CATEGORIES:
85
- "densepose_lvis_v1_ds2_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_ds2_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
- CLASS_TO_MESH_NAME_MAPPING:
106
- "0": "bear_4936"
107
- "1": "cow_5002"
108
- "2": "cat_7466"
109
- "3": "dog_7466"
110
- "4": "elephant_5002"
111
- "5": "giraffe_5002"
112
- "6": "horse_5004"
113
- "7": "sheep_5004"
114
- "8": "zebra_5002"
115
- SOLVER:
116
- MAX_ITER: 16000
117
- STEPS: (12000, 14000)
118
- DENSEPOSE_EVALUATION:
119
- EVALUATE_MESH_ALIGNMENT: True