add SMPL mesh to m2m cycle loss config
Summary: The presence of SMPL mesh embeddings is important for M2M cycle loss. Added SMPL explicitly
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
- WEIGHTS: "https://dl.fbaipublicfiles.com/densepose/cse/densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k/270668502/model_final_21b1d2.pkl"
+ 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:
@@ -78,6 +78,13 @@ MODEL:
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"
@@ -119,3 +126,13 @@ SOLVER:
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
