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
|