Dongyang Jin
2c29afadf3
* pose * pose * pose * pose * 你的提交消息 * pose * pose * Delete train1.sh * pretreatment * configs * pose * reference * Update gaittr.py * naming * naming * Update transform.py * update for datasets * update README * update name and README * update * Update transform.py
101 lines
1.9 KiB
YAML
101 lines
1.9 KiB
YAML
data_cfg:
|
|
dataset_name: Gait3D
|
|
dataset_root: your_path
|
|
dataset_partition: ./datasets/Gait3D/Gait3D.json
|
|
test_dataset_name: Gait3D
|
|
num_workers: 8
|
|
remove_no_gallery: false
|
|
frame_threshold: 24
|
|
|
|
evaluator_cfg:
|
|
enable_float16: false
|
|
restore_ckpt_strict: true
|
|
restore_hint: 20000
|
|
save_name: GaitGraph1_phase1
|
|
sampler:
|
|
batch_size: 256
|
|
frames_num_fixed: 130
|
|
frames_num_max: 50
|
|
frames_num_min: 25
|
|
sample_type: fixed_ordered
|
|
frames_skip_num: 0
|
|
metric: euc
|
|
eval_func: evaluate_Gait3D
|
|
transform:
|
|
- type: Compose
|
|
trf_cfg:
|
|
- type: SelectSequenceCenter
|
|
sequence_length: 24
|
|
- type: GaitGraph1Input
|
|
|
|
loss_cfg:
|
|
- loss_term_weight: 1
|
|
temperature: 0.01
|
|
type: SupConLoss_Re
|
|
log_prefix: SupConLoss
|
|
|
|
model_cfg:
|
|
model: GaitGraph1
|
|
joint_format: coco
|
|
input_num: 1
|
|
reduction: 8
|
|
block: Bottleneck # Basic, initial
|
|
input_branch:
|
|
- 3
|
|
- 64
|
|
- 64
|
|
- 32
|
|
main_stream:
|
|
- 32
|
|
- 128
|
|
- 128
|
|
- 256
|
|
- 256
|
|
num_class: 128
|
|
tta: true
|
|
|
|
optimizer_cfg:
|
|
lr: 0.01
|
|
solver: Adam
|
|
weight_decay: 0.00001
|
|
|
|
scheduler_cfg:
|
|
max_lr: 0.01
|
|
total_steps: 20000
|
|
scheduler: OneCycleLR
|
|
|
|
trainer_cfg:
|
|
enable_float16: false
|
|
log_iter: 100
|
|
with_test: true
|
|
restore_ckpt_strict: false
|
|
restore_hint: 0
|
|
save_iter: 500
|
|
save_name: GaitGraph1_phase1
|
|
sync_BN: true
|
|
total_iter: 20000
|
|
sampler:
|
|
batch_shuffle: true
|
|
frames_num_fixed: 130
|
|
frames_num_max: 50
|
|
frames_num_min: 25
|
|
sample_type: fixed_ordered #Repeat sample
|
|
frames_skip_num: 0
|
|
batch_size: 128
|
|
type: CommonSampler
|
|
transform:
|
|
- type: TwoView
|
|
trf_cfg:
|
|
- type: MirrorPoses
|
|
probability: 0.5
|
|
- type: FlipSequence
|
|
probability: 0.5
|
|
- type: RandomSelectSequence
|
|
sequence_length: 16
|
|
- type: PointNoise
|
|
std: 0.05
|
|
- type: JointNoise
|
|
std: 0.1
|
|
- type: GaitGraph1Input
|
|
|