Skip to content

GitLab

Commit b634945d authored by limm's avatar limm
Browse files

support v0.6

parent 5b3792fc
Hide whitespace changes
Inline Side-by-side
Showing with 1223 additions and 0 deletions
projects/DensePose/configs/cse/Base-DensePose-RCNN-FPN-Human.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
ROI_DENSEPOSE_HEAD:
CSE:
EMBEDDERS:
"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_coco_2014_train_cse"
- "densepose_coco_2014_valminusminival_cse"
TEST:
- "densepose_coco_2014_minival_cse"
CLASS_TO_MESH_NAME_MAPPING:
"0": "smpl_27554"
projects/DensePose/configs/cse/Base-DensePose-RCNN-FPN.yaml 0 → 100644
View file @ b634945d
VERSION: 2
MODEL:
META_ARCHITECTURE: "GeneralizedRCNN"
BACKBONE:
NAME: "build_resnet_fpn_backbone"
RESNETS:
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
FPN:
IN_FEATURES: ["res2", "res3", "res4", "res5"]
ANCHOR_GENERATOR:
SIZES: [[32], [64], [128], [256], [512]] # One size for each in feature map
ASPECT_RATIOS: [[0.5, 1.0, 2.0]] # Three aspect ratios (same for all in feature maps)
RPN:
IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
PRE_NMS_TOPK_TRAIN: 2000 # Per FPN level
PRE_NMS_TOPK_TEST: 1000 # Per FPN level
# Detectron1 uses 2000 proposals per-batch,
# (See "modeling/rpn/rpn_outputs.py" for details of this legacy issue)
# which is approximately 1000 proposals per-image since the default batch size for FPN is 2.
POST_NMS_TOPK_TRAIN: 1000
POST_NMS_TOPK_TEST: 1000
DENSEPOSE_ON: True
ROI_HEADS:
NAME: "DensePoseROIHeads"
IN_FEATURES: ["p2", "p3", "p4", "p5"]
NUM_CLASSES: 1
ROI_BOX_HEAD:
NAME: "FastRCNNConvFCHead"
NUM_FC: 2
POOLER_RESOLUTION: 7
POOLER_SAMPLING_RATIO: 2
POOLER_TYPE: "ROIAlign"
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
POOLER_TYPE: "ROIAlign"
NUM_COARSE_SEGM_CHANNELS: 2
PREDICTOR_NAME: "DensePoseEmbeddingPredictor"
LOSS_NAME: "DensePoseCseLoss"
CSE:
# embedding loss, possible values:
# - "EmbeddingLoss"
# - "SoftEmbeddingLoss"
EMBED_LOSS_NAME: "EmbeddingLoss"
SOLVER:
IMS_PER_BATCH: 16
BASE_LR: 0.01
STEPS: (60000, 80000)
MAX_ITER: 90000
WARMUP_FACTOR: 0.1
CLIP_GRADIENTS:
CLIP_TYPE: norm
CLIP_VALUE: 1.0
ENABLED: true
NORM_TYPE: 2.0
INPUT:
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
DENSEPOSE_EVALUATION:
TYPE: cse
STORAGE: file
projects/DensePose/configs/cse/densepose_rcnn_R_101_FPN_DL_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
CSE:
EMBED_LOSS_NAME: "EmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_101_FPN_DL_soft_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_101_FPN_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "EmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_101_FPN_soft_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_DL_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
CSE:
EMBED_LOSS_NAME: "EmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "EmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_16k.yaml 0 → 100644
View file @ b634945d
_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_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"
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
CATEGORY_MAPS:
"densepose_lvis_v1_ds2_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_ds2_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: 16000
STEPS: (12000, 14000)
DENSEPOSE_EVALUATION:
EVALUATE_MESH_ALIGNMENT: True
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml 0 → 100644
View file @ b634945d
_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
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_16k.yaml 0 → 100644
View file @ b634945d
_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"
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
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"
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
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_i2m_16k.yaml 0 → 100644
View file @ b634945d
_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"
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
PIX_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"
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
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_m2m_16k.yaml 0 → 100644
View file @ b634945d
_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
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml 0 → 100644
View file @ b634945d
_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: 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
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"
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
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_finetune_4k.yaml 0 → 100644
View file @ b634945d
_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: 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
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
CLASS_TO_MESH_NAME_MAPPING:
"0": "bear_4936"
"1": "cow_5002"
"2": "cat_5001"
"3": "dog_5002"
"4": "elephant_5002"
"5": "giraffe_5002"
"6": "horse_5004"
"7": "sheep_5004"
"8": "zebra_5002"
SOLVER:
MAX_ITER: 4000
STEPS: (3000, 3500)
DENSEPOSE_EVALUATION:
EVALUATE_MESH_ALIGNMENT: True
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k.yaml 0 → 100644
View file @ b634945d
_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: 9
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
COARSE_SEGM_TRAINED_BY_MASKS: True
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
EMBED_LOSS_WEIGHT: 0.0
EMBEDDING_DIST_GAUSS_SIGMA: 0.1
GEODESIC_DIST_GAUSS_SIGMA: 0.1
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"
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: 24000
STEPS: (20000, 22000)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_chimps_finetune_4k.yaml 0 → 100644
View file @ b634945d
_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_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
EMBEDDING_DIST_GAUSS_SIGMA: 0.1
GEODESIC_DIST_GAUSS_SIGMA: 0.1
EMBEDDERS:
"chimp_5029":
TYPE: vertex_feature
NUM_VERTICES: 5029
FEATURE_DIM: 256
FEATURES_TRAINABLE: False
IS_TRAINABLE: True
INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_chimp_5029_256.pkl"
DATASETS:
TRAIN:
- "densepose_chimps_cse_train"
TEST:
- "densepose_chimps_cse_val"
CLASS_TO_MESH_NAME_MAPPING:
"0": "chimp_5029"
SOLVER:
MAX_ITER: 4000
STEPS: (3000, 3500)
projects/DensePose/configs/cse/densepose_rcnn_R_50_FPN_soft_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
projects/DensePose/configs/densepose_rcnn_R_101_FPN_DL_WC1M_s1x.yaml 0 → 100644
View file @ b634945d
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
UV_CONFIDENCE:
ENABLED: True
TYPE: "iid_iso"
SEGM_CONFIDENCE:
ENABLED: True
POINT_REGRESSION_WEIGHTS: 0.0005
SOLVER:
CLIP_GRADIENTS:
ENABLED: True
MAX_ITER: 130000
STEPS: (100000, 120000)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment