n-poulsen commited on
Commit
c41108a
·
verified ·
1 Parent(s): 2e0e8b8

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
- - b0
83
- - b1
84
- - b2
85
- - b3
86
- - b4
87
- - b5
88
- - b6
89
- - b7
90
- - b8
91
- - b9
92
- - b10
93
- - b11
94
- - b12
95
- - b13
96
- - b14
97
- - b15
98
- - b16
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