Hugging Face's logo

camenduru
/
HandRefiner

Model card Files
xet
Community
camenduru commited on
Commit
c7a645e
·
1 Parent(s): 58129ed

thanks to microsoft ❤

Browse files
Files changed (7) hide show
  1. graphormer_release/graphormer_3dpw_state_dict.bin +3 -0
  2. graphormer_release/graphormer_h36m_state_dict.bin +3 -0
  3. graphormer_release/graphormer_hand_state_dict.bin +3 -0
  4. hrnet/cls_hrnet_w40_sgd_lr5e-2_wd1e-4_bs32_x100.yaml +92 -0
  5. hrnet/cls_hrnet_w64_sgd_lr5e-2_wd1e-4_bs32_x100.yaml +92 -0
  6. hrnet/hrnetv2_w40_imagenet_pretrained.pth +3 -0
  7. hrnet/hrnetv2_w64_imagenet_pretrained.pth +3 -0
graphormer_release/graphormer_3dpw_state_dict.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26797a4e4b5fc08480a187834a158cadedaa44a80468409b051554b01b4dcef9
3
+ size 905842355
graphormer_release/graphormer_h36m_state_dict.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a48d64f8847343d517c885550b461782c38bba65a6a2865ddbc5f864c25eb3b3
3
+ size 905842355
graphormer_release/graphormer_hand_state_dict.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b78081d105dc730c179eb0e2366bea999ceb4363070bb3aa1ae31fcf878a8a41
3
+ size 855658184
hrnet/cls_hrnet_w40_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ GPUS: (0,1,2,3)
2
+ LOG_DIR: 'log/'
3
+ DATA_DIR: ''
4
+ OUTPUT_DIR: 'output/'
5
+ WORKERS: 4
6
+ PRINT_FREQ: 1000
7
+
8
+ MODEL:
9
+ NAME: cls_hrnet
10
+ IMAGE_SIZE:
11
+ - 224
12
+ - 224
13
+ EXTRA:
14
+ STAGE1:
15
+ NUM_MODULES: 1
16
+ NUM_RANCHES: 1
17
+ BLOCK: BOTTLENECK
18
+ NUM_BLOCKS:
19
+ - 4
20
+ NUM_CHANNELS:
21
+ - 64
22
+ FUSE_METHOD: SUM
23
+ STAGE2:
24
+ NUM_MODULES: 1
25
+ NUM_BRANCHES: 2
26
+ BLOCK: BASIC
27
+ NUM_BLOCKS:
28
+ - 4
29
+ - 4
30
+ NUM_CHANNELS:
31
+ - 40
32
+ - 80
33
+ FUSE_METHOD: SUM
34
+ STAGE3:
35
+ NUM_MODULES: 4
36
+ NUM_BRANCHES: 3
37
+ BLOCK: BASIC
38
+ NUM_BLOCKS:
39
+ - 4
40
+ - 4
41
+ - 4
42
+ NUM_CHANNELS:
43
+ - 40
44
+ - 80
45
+ - 160
46
+ FUSE_METHOD: SUM
47
+ STAGE4:
48
+ NUM_MODULES: 3
49
+ NUM_BRANCHES: 4
50
+ BLOCK: BASIC
51
+ NUM_BLOCKS:
52
+ - 4
53
+ - 4
54
+ - 4
55
+ - 4
56
+ NUM_CHANNELS:
57
+ - 40
58
+ - 80
59
+ - 160
60
+ - 320
61
+ FUSE_METHOD: SUM
62
+ CUDNN:
63
+ BENCHMARK: true
64
+ DETERMINISTIC: false
65
+ ENABLED: true
66
+ DATASET:
67
+ DATASET: 'imagenet'
68
+ DATA_FORMAT: 'jpg'
69
+ ROOT: 'data/imagenet/'
70
+ TEST_SET: 'val'
71
+ TRAIN_SET: 'train'
72
+ TEST:
73
+ BATCH_SIZE_PER_GPU: 32
74
+ MODEL_FILE: ''
75
+ TRAIN:
76
+ BATCH_SIZE_PER_GPU: 32
77
+ BEGIN_EPOCH: 0
78
+ END_EPOCH: 100
79
+ RESUME: true
80
+ LR_FACTOR: 0.1
81
+ LR_STEP:
82
+ - 30
83
+ - 60
84
+ - 90
85
+ OPTIMIZER: sgd
86
+ LR: 0.05
87
+ WD: 0.0001
88
+ MOMENTUM: 0.9
89
+ NESTEROV: true
90
+ SHUFFLE: true
91
+ DEBUG:
92
+ DEBUG: false
hrnet/cls_hrnet_w64_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ GPUS: (0,1,2,3)
2
+ LOG_DIR: 'log/'
3
+ DATA_DIR: ''
4
+ OUTPUT_DIR: 'output/'
5
+ WORKERS: 4
6
+ PRINT_FREQ: 1000
7
+
8
+ MODEL:
9
+ NAME: cls_hrnet
10
+ IMAGE_SIZE:
11
+ - 224
12
+ - 224
13
+ EXTRA:
14
+ STAGE1:
15
+ NUM_MODULES: 1
16
+ NUM_RANCHES: 1
17
+ BLOCK: BOTTLENECK
18
+ NUM_BLOCKS:
19
+ - 4
20
+ NUM_CHANNELS:
21
+ - 64
22
+ FUSE_METHOD: SUM
23
+ STAGE2:
24
+ NUM_MODULES: 1
25
+ NUM_BRANCHES: 2
26
+ BLOCK: BASIC
27
+ NUM_BLOCKS:
28
+ - 4
29
+ - 4
30
+ NUM_CHANNELS:
31
+ - 64
32
+ - 128
33
+ FUSE_METHOD: SUM
34
+ STAGE3:
35
+ NUM_MODULES: 4
36
+ NUM_BRANCHES: 3
37
+ BLOCK: BASIC
38
+ NUM_BLOCKS:
39
+ - 4
40
+ - 4
41
+ - 4
42
+ NUM_CHANNELS:
43
+ - 64
44
+ - 128
45
+ - 256
46
+ FUSE_METHOD: SUM
47
+ STAGE4:
48
+ NUM_MODULES: 3
49
+ NUM_BRANCHES: 4
50
+ BLOCK: BASIC
51
+ NUM_BLOCKS:
52
+ - 4
53
+ - 4
54
+ - 4
55
+ - 4
56
+ NUM_CHANNELS:
57
+ - 64
58
+ - 128
59
+ - 256
60
+ - 512
61
+ FUSE_METHOD: SUM
62
+ CUDNN:
63
+ BENCHMARK: true
64
+ DETERMINISTIC: false
65
+ ENABLED: true
66
+ DATASET:
67
+ DATASET: 'imagenet'
68
+ DATA_FORMAT: 'jpg'
69
+ ROOT: 'data/imagenet/'
70
+ TEST_SET: 'val'
71
+ TRAIN_SET: 'train'
72
+ TEST:
73
+ BATCH_SIZE_PER_GPU: 32
74
+ MODEL_FILE: ''
75
+ TRAIN:
76
+ BATCH_SIZE_PER_GPU: 32
77
+ BEGIN_EPOCH: 0
78
+ END_EPOCH: 100
79
+ RESUME: true
80
+ LR_FACTOR: 0.1
81
+ LR_STEP:
82
+ - 30
83
+ - 60
84
+ - 90
85
+ OPTIMIZER: sgd
86
+ LR: 0.05
87
+ WD: 0.0001
88
+ MOMENTUM: 0.9
89
+ NESTEROV: true
90
+ SHUFFLE: true
91
+ DEBUG:
92
+ DEBUG: false
hrnet/hrnetv2_w40_imagenet_pretrained.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed0b031c313fa20572280cb293bfb28eaf4ddf450c32d47b2792acc33abd696d
3
+ size 230939026
hrnet/hrnetv2_w64_imagenet_pretrained.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41ed675bcd1f4f4b62a49bad64901f08f8b67ed744b715da87738f926dae685c
3
+ size 513111608