Hugging Face's logo

Spaces:
ake178178
/
IDM-VTON-dedao-demo01
Runtime error

App Files Community
IDM-VTON-dedao-demo01
File size: 3,809 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
 
_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)