File size: 4,370 Bytes
938e515 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 |
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "https://dl.fbaipublicfiles.com/densepose/cse/densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k/267687159/model_final_354e61.pkl"
RESNETS:
DEPTH: 50
ROI_HEADS:
NUM_CLASSES: 9
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
COARSE_SEGM_TRAINED_BY_MASKS: True
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
EMBEDDING_DIST_GAUSS_SIGMA: 0.1
GEODESIC_DIST_GAUSS_SIGMA: 0.1
SHAPE_TO_SHAPE_CYCLE_LOSS:
ENABLED: True
EMBEDDERS:
"cat_7466":
TYPE: vertex_feature
NUM_VERTICES: 7466
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_cat_7466_256.pkl"
"dog_7466":
TYPE: vertex_feature
NUM_VERTICES: 7466
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_dog_7466_256.pkl"
"sheep_5004":
TYPE: vertex_feature
NUM_VERTICES: 5004
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_sheep_5004_256.pkl"
"horse_5004":
TYPE: vertex_feature
NUM_VERTICES: 5004
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_horse_5004_256.pkl"
"zebra_5002":
TYPE: vertex_feature
NUM_VERTICES: 5002
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_zebra_5002_256.pkl"
"giraffe_5002":
TYPE: vertex_feature
NUM_VERTICES: 5002
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_giraffe_5002_256.pkl"
"elephant_5002":
TYPE: vertex_feature
NUM_VERTICES: 5002
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_elephant_5002_256.pkl"
"cow_5002":
TYPE: vertex_feature
NUM_VERTICES: 5002
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_cow_5002_256.pkl"
"bear_4936":
TYPE: vertex_feature
NUM_VERTICES: 4936
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_bear_4936_256.pkl"
"smpl_27554":
TYPE: vertex_feature
NUM_VERTICES: 27554
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_smpl_27554_256.pkl"
DATASETS:
TRAIN:
- "densepose_lvis_v1_ds2_train_v1"
TEST:
- "densepose_lvis_v1_ds2_val_v1"
WHITELISTED_CATEGORIES:
"densepose_lvis_v1_ds2_train_v1":
- 943 # sheep
- 1202 # zebra
- 569 # horse
- 496 # giraffe
- 422 # elephant
- 80 # cow
- 76 # bear
- 225 # cat
- 378 # dog
"densepose_lvis_v1_ds2_val_v1":
- 943 # sheep
- 1202 # zebra
- 569 # horse
- 496 # giraffe
- 422 # elephant
- 80 # cow
- 76 # bear
- 225 # cat
- 378 # dog
CLASS_TO_MESH_NAME_MAPPING:
"0": "bear_4936"
"1": "cow_5002"
"2": "cat_7466"
"3": "dog_7466"
"4": "elephant_5002"
"5": "giraffe_5002"
"6": "horse_5004"
"7": "sheep_5004"
"8": "zebra_5002"
SOLVER:
MAX_ITER: 16000
STEPS: (12000, 14000)
DENSEPOSE_EVALUATION:
EVALUATE_MESH_ALIGNMENT: True
MESH_ALIGNMENT_MESH_NAMES:
- bear_4936
- cow_5002
- cat_7466
- dog_7466
- elephant_5002
- giraffe_5002
- horse_5004
- sheep_5004
- zebra_5002
|