Spaces:
Runtime error
Runtime error
IDM-VTON-dedao-demo01
/
configs
/quick_schedules
/cse
/densepose_rcnn_R_50_FPN_soft_animals_finetune_instant_test.yaml
_BASE_: "../../cse/Base-DensePose-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
RESNETS: | |
DEPTH: 50 | |
ROI_HEADS: | |
NUM_CLASSES: 9 | |
ROI_DENSEPOSE_HEAD: | |
NAME: "DensePoseV1ConvXHead" | |
CSE: | |
EMBED_LOSS_NAME: "SoftEmbeddingLoss" | |
EMBEDDING_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_train1" | |
- "densepose_lvis_v1_train2" | |
TEST: | |
- "densepose_lvis_v1_val_animals_100" | |
WHITELISTED_CATEGORIES: | |
"densepose_lvis_v1_train1": | |
- 943 # sheep | |
- 1202 # zebra | |
- 569 # horse | |
- 496 # giraffe | |
- 422 # elephant | |
- 80 # cow | |
- 76 # bear | |
- 225 # cat | |
- 378 # dog | |
"densepose_lvis_v1_train2": | |
- 943 # sheep | |
- 1202 # zebra | |
- 569 # horse | |
- 496 # giraffe | |
- 422 # elephant | |
- 80 # cow | |
- 76 # bear | |
- 225 # cat | |
- 378 # dog | |
"densepose_lvis_v1_val_animals_100": | |
- 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: 40 | |
STEPS: (30,) | |