File size: 4,467 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 |
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "https://dl.fbaipublicfiles.com/densepose/cse/densepose_rcnn_R_50_FPN_soft_s1x/250533982/model_final_2c4512.pkl"
RESNETS:
DEPTH: 50
ROI_HEADS:
NUM_CLASSES: 1
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
EMBEDDERS:
"cat_5001":
TYPE: vertex_feature
NUM_VERTICES: 5001
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_cat_5001_256.pkl"
"dog_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_dog_5002_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"
DATASETS:
TRAIN:
- "densepose_lvis_v1_ds1_train_v1"
TEST:
- "densepose_lvis_v1_ds1_val_v1"
WHITELISTED_CATEGORIES:
"densepose_lvis_v1_ds1_train_v1":
- 943 # sheep
- 1202 # zebra
- 569 # horse
- 496 # giraffe
- 422 # elephant
- 80 # cow
- 76 # bear
- 225 # cat
- 378 # dog
"densepose_lvis_v1_ds1_val_v1":
- 943 # sheep
- 1202 # zebra
- 569 # horse
- 496 # giraffe
- 422 # elephant
- 80 # cow
- 76 # bear
- 225 # cat
- 378 # dog
CATEGORY_MAPS:
"densepose_lvis_v1_ds1_train_v1":
"1202": 943 # zebra -> sheep
"569": 943 # horse -> sheep
"496": 943 # giraffe -> sheep
"422": 943 # elephant -> sheep
"80": 943 # cow -> sheep
"76": 943 # bear -> sheep
"225": 943 # cat -> sheep
"378": 943 # dog -> sheep
"densepose_lvis_v1_ds1_val_v1":
"1202": 943 # zebra -> sheep
"569": 943 # horse -> sheep
"496": 943 # giraffe -> sheep
"422": 943 # elephant -> sheep
"80": 943 # cow -> sheep
"76": 943 # bear -> sheep
"225": 943 # cat -> sheep
"378": 943 # dog -> sheep
CLASS_TO_MESH_NAME_MAPPING:
# Note: different classes are mapped to a single class
# mesh is chosen based on GT data, so this is just some
# value which has no particular meaning
"0": "sheep_5004"
SOLVER:
MAX_ITER: 4000
STEPS: (3000, 3500)
DENSEPOSE_EVALUATION:
EVALUATE_MESH_ALIGNMENT: True
|