Upload rtmpose-x_simcc-body7_pytorch_config.yaml
Browse files
rtmpose-x_simcc-body7_pytorch_config.yaml
CHANGED
@@ -22,80 +22,29 @@ data:
|
|
22 |
top_down_crop:
|
23 |
width: 288
|
24 |
height: 384
|
25 |
-
detector:
|
26 |
-
data:
|
27 |
-
colormode: RGB
|
28 |
-
inference:
|
29 |
-
normalize_images: true
|
30 |
-
train:
|
31 |
-
affine:
|
32 |
-
p: 0.5
|
33 |
-
rotation: 30
|
34 |
-
scaling:
|
35 |
-
- 1.0
|
36 |
-
- 1.0
|
37 |
-
translation: 40
|
38 |
-
collate:
|
39 |
-
type: ResizeFromDataSizeCollate
|
40 |
-
min_scale: 0.4
|
41 |
-
max_scale: 1.0
|
42 |
-
min_short_side: 128
|
43 |
-
max_short_side: 1152
|
44 |
-
multiple_of: 32
|
45 |
-
to_square: false
|
46 |
-
hflip: true
|
47 |
-
normalize_images: true
|
48 |
-
device: auto
|
49 |
-
model:
|
50 |
-
type: SSDLite
|
51 |
-
freeze_bn_stats: true
|
52 |
-
freeze_bn_weights: false
|
53 |
-
runner:
|
54 |
-
type: DetectorTrainingRunner
|
55 |
-
eval_interval: 1
|
56 |
-
optimizer:
|
57 |
-
type: AdamW
|
58 |
-
params:
|
59 |
-
lr: 0.0001
|
60 |
-
scheduler:
|
61 |
-
type: LRListScheduler
|
62 |
-
params:
|
63 |
-
milestones:
|
64 |
-
- 160
|
65 |
-
lr_list:
|
66 |
-
- - 1e-05
|
67 |
-
snapshots:
|
68 |
-
max_snapshots: 5
|
69 |
-
save_epochs: 25
|
70 |
-
save_optimizer_state: false
|
71 |
-
train_settings:
|
72 |
-
batch_size: 16
|
73 |
-
dataloader_workers: 0
|
74 |
-
dataloader_pin_memory: false
|
75 |
-
display_iters: 500
|
76 |
-
epochs: 250
|
77 |
device: auto
|
78 |
metadata:
|
79 |
project_path: null
|
80 |
pose_config_path: rtmpose_x_body7_pytorch_config.yaml
|
81 |
bodyparts:
|
82 |
-
-
|
83 |
-
-
|
84 |
-
-
|
85 |
-
-
|
86 |
-
-
|
87 |
-
-
|
88 |
-
-
|
89 |
-
-
|
90 |
-
-
|
91 |
-
-
|
92 |
-
-
|
93 |
-
-
|
94 |
-
-
|
95 |
-
-
|
96 |
-
-
|
97 |
-
-
|
98 |
-
-
|
99 |
unique_bodyparts: []
|
100 |
individuals:
|
101 |
- idv0
|
|
|
22 |
top_down_crop:
|
23 |
width: 288
|
24 |
height: 384
|
25 |
+
detector: null
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
device: auto
|
27 |
metadata:
|
28 |
project_path: null
|
29 |
pose_config_path: rtmpose_x_body7_pytorch_config.yaml
|
30 |
bodyparts:
|
31 |
+
- nose
|
32 |
+
- left_eye
|
33 |
+
- right_eye
|
34 |
+
- left_ear
|
35 |
+
- right_ear
|
36 |
+
- left_shoulder
|
37 |
+
- right_shoulder
|
38 |
+
- left_elbow
|
39 |
+
- right_elbow
|
40 |
+
- left_wrist
|
41 |
+
- right_wrist
|
42 |
+
- left_hip
|
43 |
+
- right_hip
|
44 |
+
- left_knee
|
45 |
+
- right_knee
|
46 |
+
- left_ankle
|
47 |
+
- right_ankle
|
48 |
unique_bodyparts: []
|
49 |
individuals:
|
50 |
- idv0
|