Delete configs/baseline directory
The 'baseline' series is abandoned since GaitBase can better serve as a baseline model for OpenGait.
This commit is contained in:
@@ -1,99 +0,0 @@
|
|||||||
data_cfg:
|
|
||||||
dataset_name: CASIA-B
|
|
||||||
dataset_root: your_path
|
|
||||||
dataset_partition: ./datasets/CASIA-B/CASIA-B.json
|
|
||||||
num_workers: 1
|
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
|
||||||
test_dataset_name: CASIA-B
|
|
||||||
|
|
||||||
evaluator_cfg:
|
|
||||||
enable_float16: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 60000
|
|
||||||
save_name: Baseline
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: false
|
|
||||||
batch_size: 16
|
|
||||||
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
|
|
||||||
metric: euc # cos
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 64
|
|
||||||
|
|
||||||
loss_cfg:
|
|
||||||
- loss_term_weight: 1.0
|
|
||||||
margin: 0.2
|
|
||||||
type: TripletLoss
|
|
||||||
log_prefix: triplet
|
|
||||||
- loss_term_weight: 0.1
|
|
||||||
scale: 16
|
|
||||||
type: CrossEntropyLoss
|
|
||||||
log_prefix: softmax
|
|
||||||
log_accuracy: true
|
|
||||||
|
|
||||||
model_cfg:
|
|
||||||
model: Baseline
|
|
||||||
backbone_cfg:
|
|
||||||
in_channels: 1
|
|
||||||
layers_cfg: # Layers configuration for automatically model construction
|
|
||||||
- BC-64
|
|
||||||
- BC-64
|
|
||||||
- M
|
|
||||||
- BC-128
|
|
||||||
- BC-128
|
|
||||||
- M
|
|
||||||
- BC-256
|
|
||||||
- BC-256
|
|
||||||
type: Plain
|
|
||||||
SeparateFCs:
|
|
||||||
in_channels: 256
|
|
||||||
out_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
SeparateBNNecks:
|
|
||||||
class_num: 74
|
|
||||||
in_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
bin_num:
|
|
||||||
- 16
|
|
||||||
- 8
|
|
||||||
- 4
|
|
||||||
- 2
|
|
||||||
- 1
|
|
||||||
|
|
||||||
optimizer_cfg:
|
|
||||||
lr: 0.1
|
|
||||||
momentum: 0.9
|
|
||||||
solver: SGD
|
|
||||||
weight_decay: 0.0005
|
|
||||||
|
|
||||||
scheduler_cfg:
|
|
||||||
gamma: 0.1
|
|
||||||
milestones: # Learning Rate Reduction at each milestones
|
|
||||||
- 20000
|
|
||||||
- 40000
|
|
||||||
scheduler: MultiStepLR
|
|
||||||
trainer_cfg:
|
|
||||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
|
||||||
fix_BN: false
|
|
||||||
log_iter: 100
|
|
||||||
with_test: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 0
|
|
||||||
save_iter: 10000
|
|
||||||
save_name: Baseline
|
|
||||||
sync_BN: true
|
|
||||||
total_iter: 60000
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: true
|
|
||||||
batch_size:
|
|
||||||
- 8 # TripletSampler, batch_size[0] indicates Number of Identity
|
|
||||||
- 16 # batch_size[1] indicates Samples sequqnce for each Identity
|
|
||||||
frames_num_fixed: 30 # fixed frames number for training
|
|
||||||
frames_num_max: 50 # max frames number for unfixed training
|
|
||||||
frames_num_min: 25 # min frames number for unfixed traing
|
|
||||||
sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
|
||||||
type: TripletSampler
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 64
|
|
||||||
@@ -1,103 +0,0 @@
|
|||||||
data_cfg:
|
|
||||||
dataset_name: GREW
|
|
||||||
dataset_root: /home/data/grew-pkl-fixed
|
|
||||||
dataset_partition: ./datasets/GREW/GREW.json
|
|
||||||
num_workers: 16
|
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
|
||||||
test_dataset_name: GREW
|
|
||||||
|
|
||||||
evaluator_cfg:
|
|
||||||
enable_float16: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 250000
|
|
||||||
save_name: Baseline
|
|
||||||
eval_func: GREW_submission # identification_real_scene # identification_GREW_submission
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: false
|
|
||||||
batch_size: 64
|
|
||||||
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
|
|
||||||
metric: euc # cos
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 64
|
|
||||||
|
|
||||||
loss_cfg:
|
|
||||||
- loss_term_weight: 1.0
|
|
||||||
margin: 0.2
|
|
||||||
type: TripletLoss
|
|
||||||
log_prefix: triplet
|
|
||||||
- loss_term_weight: 0.1
|
|
||||||
scale: 16
|
|
||||||
type: CrossEntropyLoss
|
|
||||||
log_prefix: softmax
|
|
||||||
log_accuracy: true
|
|
||||||
|
|
||||||
model_cfg:
|
|
||||||
model: Baseline
|
|
||||||
backbone_cfg:
|
|
||||||
in_channels: 1
|
|
||||||
layers_cfg: # Layers configuration for automatically model construction
|
|
||||||
- BC-32
|
|
||||||
- BC-32
|
|
||||||
- M
|
|
||||||
- BC-64
|
|
||||||
- BC-64
|
|
||||||
- M
|
|
||||||
- BC-128
|
|
||||||
- BC-128
|
|
||||||
- BC-256
|
|
||||||
- BC-256
|
|
||||||
# - M
|
|
||||||
# - BC-512
|
|
||||||
# - BC-512
|
|
||||||
type: Plain
|
|
||||||
SeparateFCs:
|
|
||||||
in_channels: 256
|
|
||||||
out_channels: 256
|
|
||||||
parts_num: 16
|
|
||||||
SeparateBNNecks:
|
|
||||||
class_num: 20000
|
|
||||||
in_channels: 256
|
|
||||||
parts_num: 16
|
|
||||||
bin_num:
|
|
||||||
- 16
|
|
||||||
# - 8
|
|
||||||
# - 4
|
|
||||||
# - 2
|
|
||||||
# - 1
|
|
||||||
|
|
||||||
optimizer_cfg:
|
|
||||||
lr: 0.1
|
|
||||||
momentum: 0.9
|
|
||||||
solver: SGD
|
|
||||||
weight_decay: 0.0005
|
|
||||||
|
|
||||||
scheduler_cfg:
|
|
||||||
gamma: 0.1
|
|
||||||
milestones: # Learning Rate Reduction at each milestones
|
|
||||||
- 150000
|
|
||||||
scheduler: MultiStepLR
|
|
||||||
trainer_cfg:
|
|
||||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
|
||||||
fix_BN: false
|
|
||||||
log_iter: 100
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 0
|
|
||||||
save_iter: 10000
|
|
||||||
save_name: Baseline
|
|
||||||
sync_BN: true
|
|
||||||
total_iter: 250000
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: true
|
|
||||||
batch_size:
|
|
||||||
- 32 # TripletSampler, batch_size[0] indicates Number of Identity
|
|
||||||
- 8 # batch_size[1] indicates Samples sequqnce for each Identity
|
|
||||||
frames_num_fixed: 30 # fixed frames number for training
|
|
||||||
frames_num_max: 50 # max frames number for unfixed training
|
|
||||||
frames_num_min: 25 # min frames number for unfixed traing
|
|
||||||
sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
|
||||||
type: TripletSampler
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 64
|
|
||||||
@@ -1,104 +0,0 @@
|
|||||||
data_cfg:
|
|
||||||
dataset_name: Gait3D
|
|
||||||
dataset_root: your_path
|
|
||||||
dataset_partition: datasets/Gait3D/Gait3D.json
|
|
||||||
data_in_use: [true, false]
|
|
||||||
num_workers: 1
|
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
|
||||||
test_dataset_name: Gait3D
|
|
||||||
|
|
||||||
evaluator_cfg:
|
|
||||||
enable_float16: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 60000
|
|
||||||
save_name: Baseline
|
|
||||||
eval_func: evaluate_Gait3D
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: false
|
|
||||||
batch_size: 16
|
|
||||||
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
|
|
||||||
metric: euc # cos euc
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
|
|
||||||
loss_cfg:
|
|
||||||
- loss_term_weight: 1.0
|
|
||||||
margin: 0.2
|
|
||||||
type: TripletLoss
|
|
||||||
log_prefix: triplet
|
|
||||||
- loss_term_weight: 0.1
|
|
||||||
scale: 16
|
|
||||||
type: CrossEntropyLoss
|
|
||||||
log_prefix: softmax
|
|
||||||
log_accuracy: true
|
|
||||||
|
|
||||||
model_cfg:
|
|
||||||
model: Baseline
|
|
||||||
backbone_cfg:
|
|
||||||
in_channels: 1
|
|
||||||
layers_cfg: # Layers configuration for automatically model construction
|
|
||||||
- BC-64
|
|
||||||
- BC-64
|
|
||||||
- M
|
|
||||||
- BC-128
|
|
||||||
- BC-128
|
|
||||||
- M
|
|
||||||
- BC-256
|
|
||||||
- BC-256
|
|
||||||
# - M
|
|
||||||
# - BC-512
|
|
||||||
# - BC-512
|
|
||||||
type: Plain
|
|
||||||
SeparateFCs:
|
|
||||||
in_channels: 256
|
|
||||||
out_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
SeparateBNNecks:
|
|
||||||
class_num: 3000
|
|
||||||
in_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
bin_num:
|
|
||||||
- 16
|
|
||||||
- 8
|
|
||||||
- 4
|
|
||||||
- 2
|
|
||||||
- 1
|
|
||||||
|
|
||||||
optimizer_cfg:
|
|
||||||
lr: 0.1
|
|
||||||
momentum: 0.9
|
|
||||||
solver: SGD
|
|
||||||
weight_decay: 0.0005
|
|
||||||
|
|
||||||
scheduler_cfg:
|
|
||||||
gamma: 0.1
|
|
||||||
milestones: # Learning Rate Reduction at each milestones
|
|
||||||
- 20000
|
|
||||||
- 40000
|
|
||||||
- 50000
|
|
||||||
scheduler: MultiStepLR
|
|
||||||
|
|
||||||
trainer_cfg:
|
|
||||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
|
||||||
fix_BN: false
|
|
||||||
log_iter: 100
|
|
||||||
with_test: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 0
|
|
||||||
save_iter: 10000
|
|
||||||
save_name: Baseline
|
|
||||||
sync_BN: true
|
|
||||||
total_iter: 60000
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: true
|
|
||||||
batch_size:
|
|
||||||
- 32 # TripletSampler, batch_size[0] indicates Number of Identity
|
|
||||||
- 4 # batch_size[1] indicates Samples sequqnce for each Identity
|
|
||||||
frames_num_fixed: 30 # fixed frames number for training
|
|
||||||
frames_num_max: 50 # max frames number for unfixed training
|
|
||||||
frames_num_min: 25 # min frames number for unfixed traing
|
|
||||||
sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
|
||||||
type: TripletSampler
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
@@ -1,101 +0,0 @@
|
|||||||
data_cfg:
|
|
||||||
dataset_name: OUMVLP
|
|
||||||
dataset_root: your_path
|
|
||||||
dataset_partition: ./datasets/OUMVLP/OUMVLP.json
|
|
||||||
num_workers: 1
|
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
|
||||||
test_dataset_name: OUMVLP
|
|
||||||
|
|
||||||
evaluator_cfg:
|
|
||||||
enable_float16: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 150000
|
|
||||||
save_name: Baseline
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: false
|
|
||||||
batch_size: 4
|
|
||||||
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
|
|
||||||
metric: euc # cos
|
|
||||||
# transform:
|
|
||||||
# - type: BaseSilCuttingTransform
|
|
||||||
# img_w: 128
|
|
||||||
|
|
||||||
loss_cfg:
|
|
||||||
- loss_term_weight: 1.0
|
|
||||||
margin: 0.2
|
|
||||||
type: TripletLoss
|
|
||||||
log_prefix: triplet
|
|
||||||
- loss_term_weight: 0.1
|
|
||||||
scale: 16
|
|
||||||
type: CrossEntropyLoss
|
|
||||||
log_prefix: softmax
|
|
||||||
log_accuracy: true
|
|
||||||
|
|
||||||
model_cfg:
|
|
||||||
model: Baseline
|
|
||||||
backbone_cfg:
|
|
||||||
in_channels: 1
|
|
||||||
layers_cfg: # Layers configuration for automatically model construction
|
|
||||||
- BC-32
|
|
||||||
- BC-32
|
|
||||||
- M
|
|
||||||
- BC-64
|
|
||||||
- BC-64
|
|
||||||
- M
|
|
||||||
- BC-128
|
|
||||||
- BC-128
|
|
||||||
- BC-256
|
|
||||||
- BC-256
|
|
||||||
type: Plain
|
|
||||||
SeparateFCs:
|
|
||||||
in_channels: 256
|
|
||||||
out_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
SeparateBNNecks:
|
|
||||||
class_num: 5153
|
|
||||||
in_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
bin_num:
|
|
||||||
- 16
|
|
||||||
- 8
|
|
||||||
- 4
|
|
||||||
- 2
|
|
||||||
- 1
|
|
||||||
|
|
||||||
optimizer_cfg:
|
|
||||||
lr: 0.1
|
|
||||||
momentum: 0.9
|
|
||||||
solver: SGD
|
|
||||||
weight_decay: 0.0005
|
|
||||||
|
|
||||||
scheduler_cfg:
|
|
||||||
gamma: 0.1
|
|
||||||
milestones: # Learning Rate Reduction at each milestones
|
|
||||||
- 50000
|
|
||||||
- 100000
|
|
||||||
scheduler: MultiStepLR
|
|
||||||
trainer_cfg:
|
|
||||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
|
||||||
fix_BN: false
|
|
||||||
log_iter: 100
|
|
||||||
with_test: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 0
|
|
||||||
save_iter: 10000
|
|
||||||
save_name: Baseline
|
|
||||||
sync_BN: true
|
|
||||||
total_iter: 150000
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: true
|
|
||||||
batch_size:
|
|
||||||
- 32 # TripletSampler, batch_size[0] indicates Number of Identity
|
|
||||||
- 16 # batch_size[1] indicates Samples sequqnce for each Identity
|
|
||||||
frames_num_fixed: 30 # fixed frames number for training
|
|
||||||
frames_num_max: 50 # max frames number for unfixed training
|
|
||||||
frames_num_min: 25 # min frames number for unfixed traing
|
|
||||||
sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
|
||||||
type: TripletSampler
|
|
||||||
# transform:
|
|
||||||
# - type: BaseSilCuttingTransform
|
|
||||||
# img_w: 128
|
|
||||||
@@ -1,98 +0,0 @@
|
|||||||
data_cfg:
|
|
||||||
dataset_name: HID
|
|
||||||
dataset_root: your_path
|
|
||||||
dataset_partition: ./datasets/HID/HID2023.json
|
|
||||||
num_workers: 1
|
|
||||||
remove_no_gallery: false # Remove probe if no gallery for it
|
|
||||||
evaluator_cfg:
|
|
||||||
enable_float16: true
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 60000
|
|
||||||
save_name: Baseline
|
|
||||||
eval_func: HID_submission
|
|
||||||
rerank: False
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: false
|
|
||||||
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
|
|
||||||
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
|
||||||
metric: euc # cos
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 128
|
|
||||||
loss_cfg:
|
|
||||||
- loss_term_weight: 1.0
|
|
||||||
margin: 0.2
|
|
||||||
type: TripletLoss
|
|
||||||
log_prefix: triplet
|
|
||||||
- loss_term_weight: 0.1
|
|
||||||
scale: 16
|
|
||||||
type: CrossEntropyLoss
|
|
||||||
log_prefix: softmax
|
|
||||||
log_accuracy: true
|
|
||||||
|
|
||||||
model_cfg:
|
|
||||||
model: Baseline
|
|
||||||
backbone_cfg:
|
|
||||||
in_channels: 1
|
|
||||||
layers_cfg: # Layers configuration for automatically model construction
|
|
||||||
- BC-64
|
|
||||||
- BC-64
|
|
||||||
- M
|
|
||||||
- BC-128
|
|
||||||
- BC-128
|
|
||||||
- M
|
|
||||||
- BC-256
|
|
||||||
- BC-256
|
|
||||||
- M
|
|
||||||
- BC-512
|
|
||||||
- BC-512
|
|
||||||
type: Plain
|
|
||||||
SeparateFCs:
|
|
||||||
in_channels: 512
|
|
||||||
out_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
SeparateBNNecks:
|
|
||||||
class_num: 500
|
|
||||||
in_channels: 256
|
|
||||||
parts_num: 31
|
|
||||||
bin_num:
|
|
||||||
- 16
|
|
||||||
- 8
|
|
||||||
- 4
|
|
||||||
- 2
|
|
||||||
- 1
|
|
||||||
|
|
||||||
optimizer_cfg:
|
|
||||||
lr: 0.1
|
|
||||||
momentum: 0.9
|
|
||||||
solver: SGD
|
|
||||||
weight_decay: 0.0005
|
|
||||||
|
|
||||||
scheduler_cfg:
|
|
||||||
gamma: 0.1
|
|
||||||
milestones: # Learning Rate Reduction at each milestones
|
|
||||||
- 20000
|
|
||||||
- 40000
|
|
||||||
scheduler: MultiStepLR
|
|
||||||
trainer_cfg:
|
|
||||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
|
||||||
fix_BN: false
|
|
||||||
log_iter: 100
|
|
||||||
restore_ckpt_strict: true
|
|
||||||
restore_hint: 0
|
|
||||||
save_iter: 10000
|
|
||||||
save_name: Baseline
|
|
||||||
sync_BN: true
|
|
||||||
total_iter: 60000
|
|
||||||
sampler:
|
|
||||||
batch_shuffle: true
|
|
||||||
batch_size:
|
|
||||||
- 16 # TripletSampler, batch_size[0] indicates Number of Identity
|
|
||||||
- 8 # batch_size[1] indicates Samples sequqnce for each Identity
|
|
||||||
frames_num_fixed: 30 # fixed frames number for training
|
|
||||||
sample_type: fixed_unordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
|
||||||
type: TripletSampler
|
|
||||||
transform:
|
|
||||||
- type: BaseSilCuttingTransform
|
|
||||||
img_w: 128
|
|
||||||
Reference in New Issue
Block a user