Update gaitbase_hid.yaml

This commit is contained in:
Dongyang Jin
2025-04-26 22:20:54 +08:00
committed by GitHub
parent 2ef2dfcef2
commit a5a9bef9fb
+5 -5
View File
@@ -1,22 +1,22 @@
data_cfg: data_cfg:
dataset_name: HID dataset_name: HID
dataset_root: your_path dataset_root: your_path
dataset_partition: ./datasets/HID/HID2023.json dataset_partition: ./datasets/HID/HID.json
num_workers: 1 num_workers: 1
remove_no_gallery: false # Remove probe if no gallery for it remove_no_gallery: false # Remove probe if no gallery for it
evaluator_cfg: evaluator_cfg:
enable_float16: true enable_float16: true
restore_ckpt_strict: true restore_ckpt_strict: true
restore_hint: 60000 restore_hint: 60000
save_name: GaitBase save_name: Baseline
eval_func: HID_submission eval_func: HID_submission
rerank: False
sampler: sampler:
batch_shuffle: false batch_shuffle: false
batch_size: 8 batch_size: 8
sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; Other options: fixed_unordered sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; Other options: fixed_unordered
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
metric: euc # cos metric: euc # cos
rerank: false
transform: transform:
- type: BaseSilCuttingTransform - type: BaseSilCuttingTransform
img_w: 128 img_w: 128
@@ -31,7 +31,6 @@ loss_cfg:
log_prefix: softmax log_prefix: softmax
log_accuracy: true log_accuracy: true
model_cfg: model_cfg:
model: Baseline model: Baseline
backbone_cfg: backbone_cfg:
@@ -64,6 +63,7 @@ model_cfg:
bin_num: bin_num:
- 16 - 16
optimizer_cfg: optimizer_cfg:
lr: 0.1 lr: 0.1
momentum: 0.9 momentum: 0.9
@@ -83,7 +83,7 @@ trainer_cfg:
restore_ckpt_strict: true restore_ckpt_strict: true
restore_hint: 0 restore_hint: 0
save_iter: 10000 save_iter: 10000
save_name: GaitBase save_name: Baseline
sync_BN: true sync_BN: true
total_iter: 60000 total_iter: 60000
sampler: sampler: