|
_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 |
|
- 1202 |
|
- 569 |
|
- 496 |
|
- 422 |
|
- 80 |
|
- 76 |
|
- 225 |
|
- 378 |
|
"densepose_lvis_v1_ds1_val_v1": |
|
- 943 |
|
- 1202 |
|
- 569 |
|
- 496 |
|
- 422 |
|
- 80 |
|
- 76 |
|
- 225 |
|
- 378 |
|
CATEGORY_MAPS: |
|
"densepose_lvis_v1_ds1_train_v1": |
|
"1202": 943 |
|
"569": 943 |
|
"496": 943 |
|
"422": 943 |
|
"80": 943 |
|
"76": 943 |
|
"225": 943 |
|
"378": 943 |
|
"densepose_lvis_v1_ds1_val_v1": |
|
"1202": 943 |
|
"569": 943 |
|
"496": 943 |
|
"422": 943 |
|
"80": 943 |
|
"76": 943 |
|
"225": 943 |
|
"378": 943 |
|
CLASS_TO_MESH_NAME_MAPPING: |
|
|
|
|
|
|
|
"0": "sheep_5004" |
|
SOLVER: |
|
MAX_ITER: 4000 |
|
STEPS: (3000, 3500) |
|
DENSEPOSE_EVALUATION: |
|
EVALUATE_MESH_ALIGNMENT: True |
|
|