release swingait
This commit is contained in:
jdyjjj
@@ -0,0 +1,100 @@
|
||||
data_cfg:
|
||||
dataset_name: CCPG
|
||||
dataset_root: /data/CCPG/Released/CCPG-end2end-pkl/
|
||||
dataset_partition: ./datasets/CCPG/CCPG.json
|
||||
num_workers: 1
|
||||
data_in_use: [True, False, False, False]
|
||||
remove_no_gallery: false # Remove probe if no gallery for it
|
||||
test_dataset_name: CCPG
|
||||
|
||||
evaluator_cfg:
|
||||
enable_float16: true
|
||||
restore_ckpt_strict: true
|
||||
restore_hint: 80000
|
||||
save_name: SwinGait3D_B1122_C2
|
||||
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
|
||||
eval_func: evaluate_CCPG
|
||||
metric: euc # cos
|
||||
transform:
|
||||
- type: BaseSilCuttingTransform
|
||||
|
||||
loss_cfg:
|
||||
- loss_term_weight: 1.0
|
||||
margin: 0.2
|
||||
type: TripletLoss
|
||||
log_prefix: triplet
|
||||
- loss_term_weight: 1.0
|
||||
scale: 16
|
||||
type: CrossEntropyLoss
|
||||
log_prefix: softmax
|
||||
log_accuracy: true
|
||||
|
||||
model_cfg:
|
||||
model: SwinGait
|
||||
Backbone:
|
||||
mode: p3d
|
||||
in_channels: 1
|
||||
layers:
|
||||
- 1
|
||||
- 1
|
||||
- 2
|
||||
- 2
|
||||
channels:
|
||||
- 64
|
||||
- 128
|
||||
bin_num:
|
||||
- 15
|
||||
SeparateBNNecks:
|
||||
in_channels: 256
|
||||
class_num: 100
|
||||
parts_num: 16
|
||||
|
||||
optimizer_cfg:
|
||||
lr: 0.0003
|
||||
solver: AdamW
|
||||
weight_decay: 0.02
|
||||
|
||||
scheduler_cfg:
|
||||
gamma: 0.1
|
||||
milestones: # Learning Rate Reduction at each milestones
|
||||
- 100000000
|
||||
scheduler: MultiStepLR
|
||||
|
||||
trainer_cfg:
|
||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
||||
fix_BN: false
|
||||
log_iter: 100
|
||||
with_test: false
|
||||
optimizer_reset: True
|
||||
restore_ckpt_strict: false
|
||||
restore_hint: /home/jdy/fanchao/OpenGait_230719/output/CCPG/DeepGaitV2/DeepGaitV2_B1111_C2_P3D/checkpoints/DeepGaitV2_B1111_C2_P3D-60000.pt
|
||||
save_iter: 20000
|
||||
save_name: SwinGait3D_B1122_C2
|
||||
sync_BN: true
|
||||
T_max_iter: 60000
|
||||
total_iter: 80000
|
||||
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_skip_num: 4
|
||||
sample_type: fixed_ordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
||||
type: TripletSampler
|
||||
transform:
|
||||
- type: Compose
|
||||
trf_cfg:
|
||||
- type: RandomPerspective
|
||||
prob: 0.2
|
||||
- type: BaseSilCuttingTransform
|
||||
- type: RandomHorizontalFlip
|
||||
prob: 0.2
|
||||
- type: RandomRotate
|
||||
prob: 0.2
|
||||
|
||||
|
||||
@@ -0,0 +1,98 @@
|
||||
data_cfg:
|
||||
dataset_name: SUSTech1K
|
||||
dataset_root: /data/SUSTech1K-Released-pkl/
|
||||
dataset_partition: ./datasets/SUSTech1K/SUSTech1K.json
|
||||
num_workers: 4
|
||||
data_in_use: [false, false, false, false, false, false, true, false, false, false, false, false, false, false, false, false]
|
||||
remove_no_gallery: false # Remove probe if no gallery for it
|
||||
test_dataset_name: SUSTech1K
|
||||
|
||||
evaluator_cfg:
|
||||
enable_float16: true
|
||||
restore_ckpt_strict: true
|
||||
restore_hint: 60000
|
||||
save_name: SwinGait3D_B1122_C2
|
||||
eval_func: evaluate_indoor_dataset #evaluate_Gait3D
|
||||
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
|
||||
|
||||
loss_cfg:
|
||||
- loss_term_weight: 1.0
|
||||
margin: 0.2
|
||||
type: TripletLoss
|
||||
log_prefix: triplet
|
||||
- loss_term_weight: 1.0
|
||||
scale: 16
|
||||
type: CrossEntropyLoss
|
||||
log_prefix: softmax
|
||||
log_accuracy: true
|
||||
|
||||
model_cfg:
|
||||
model: SwinGait
|
||||
Backbone:
|
||||
mode: p3d
|
||||
in_channels: 1
|
||||
layers:
|
||||
- 1
|
||||
- 1
|
||||
- 2
|
||||
- 2
|
||||
channels:
|
||||
- 64
|
||||
- 128
|
||||
bin_num:
|
||||
- 15
|
||||
SeparateBNNecks:
|
||||
in_channels: 256
|
||||
class_num: 250
|
||||
parts_num: 16
|
||||
|
||||
optimizer_cfg:
|
||||
lr: 0.0003
|
||||
solver: AdamW
|
||||
weight_decay: 0.02
|
||||
|
||||
scheduler_cfg:
|
||||
gamma: 0.1
|
||||
milestones: # Learning Rate Reduction at each milestones
|
||||
- 100000000
|
||||
scheduler: MultiStepLR
|
||||
|
||||
trainer_cfg:
|
||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
||||
fix_BN: false
|
||||
with_test: true #true
|
||||
log_iter: 100
|
||||
optimizer_reset: True
|
||||
restore_ckpt_strict: false
|
||||
restore_hint: /home/jdy/fanchao/OpenGait_230719/output/SUSTech1K/DeepGaitV2/DeepGaitV2_B1111_C2_P3D/checkpoints/DeepGaitV2_B1111_C2_P3D-50000.pt
|
||||
save_iter: 10000
|
||||
save_name: SwinGait3D_B1122_C2
|
||||
sync_BN: true
|
||||
T_max_iter: 50000
|
||||
total_iter: 60000
|
||||
sampler:
|
||||
batch_shuffle: true
|
||||
batch_size:
|
||||
- 8 # TripletSampler, batch_size[0] indicates Number of Identity
|
||||
- 8 # batch_size[1] indicates Samples sequqnce for each Identity
|
||||
frames_num_fixed: 10 # 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: Compose
|
||||
trf_cfg:
|
||||
- type: RandomPerspective
|
||||
prob: 0.2
|
||||
- type: BaseSilCuttingTransform
|
||||
- type: RandomHorizontalFlip
|
||||
prob: 0.2
|
||||
- type: RandomRotate
|
||||
prob: 0.2
|
||||
|
||||
@@ -0,0 +1,99 @@
|
||||
data_cfg:
|
||||
dataset_name: GREW
|
||||
dataset_root: your_path
|
||||
dataset_partition: ./datasets/GREW/GREW.json
|
||||
num_workers: 1
|
||||
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: 200000
|
||||
save_name: SwinGait3D_B1442_C2
|
||||
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
|
||||
eval_func: GREW_submission
|
||||
metric: euc # cos
|
||||
transform:
|
||||
- type: BaseSilCuttingTransform
|
||||
|
||||
loss_cfg:
|
||||
- loss_term_weight: 1.0
|
||||
margin: 0.2
|
||||
type: TripletLoss
|
||||
log_prefix: triplet
|
||||
- loss_term_weight: 1.0
|
||||
scale: 16
|
||||
type: CrossEntropyLoss
|
||||
log_prefix: softmax
|
||||
log_accuracy: true
|
||||
|
||||
model_cfg:
|
||||
model: SwinGait
|
||||
Backbone:
|
||||
mode: p3d
|
||||
in_channels: 1
|
||||
layers:
|
||||
- 1
|
||||
- 4
|
||||
- 4
|
||||
- 2
|
||||
channels:
|
||||
- 64
|
||||
- 128
|
||||
bin_num:
|
||||
- 15
|
||||
SeparateBNNecks:
|
||||
in_channels: 256
|
||||
class_num: 20000
|
||||
parts_num: 16
|
||||
|
||||
optimizer_cfg:
|
||||
lr: 0.0003
|
||||
solver: AdamW
|
||||
weight_decay: 0.02
|
||||
|
||||
scheduler_cfg:
|
||||
gamma: 0.1
|
||||
milestones: # Learning Rate Reduction at each milestones
|
||||
- 100000000
|
||||
scheduler: MultiStepLR
|
||||
|
||||
trainer_cfg:
|
||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
||||
fix_BN: false
|
||||
log_iter: 100
|
||||
with_test: false
|
||||
optimizer_reset: True
|
||||
restore_ckpt_strict: false
|
||||
restore_hint: /home/jdy/fanchao/OpenGait_230719/output/GREW/DeepGaitV2/DeepGaitV2_B1441_C2_P3D/checkpoints/DeepGaitV2_B1441_C2_P3D-180000.pt
|
||||
save_iter: 20000
|
||||
save_name: SwinGait3D_B1442_C2
|
||||
sync_BN: true
|
||||
T_max_iter: 150000
|
||||
total_iter: 200000
|
||||
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_skip_num: 4
|
||||
sample_type: fixed_ordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
||||
type: TripletSampler
|
||||
transform:
|
||||
- type: Compose
|
||||
trf_cfg:
|
||||
- type: RandomPerspective
|
||||
prob: 0.2
|
||||
- type: BaseSilCuttingTransform
|
||||
- type: RandomHorizontalFlip
|
||||
prob: 0.2
|
||||
- type: RandomRotate
|
||||
prob: 0.2
|
||||
|
||||
|
||||
@@ -0,0 +1,99 @@
|
||||
data_cfg:
|
||||
dataset_name: Gait3D
|
||||
dataset_root: your_path
|
||||
dataset_partition: ./datasets/Gait3D/Gait3D.json
|
||||
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: 80000
|
||||
save_name: SwinGait3D_B1442_C2
|
||||
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
|
||||
eval_func: evaluate_Gait3D
|
||||
metric: euc # cos
|
||||
transform:
|
||||
- type: BaseSilTransform
|
||||
|
||||
loss_cfg:
|
||||
- loss_term_weight: 1.0
|
||||
margin: 0.2
|
||||
type: TripletLoss
|
||||
log_prefix: triplet
|
||||
- loss_term_weight: 1.0
|
||||
scale: 16
|
||||
type: CrossEntropyLoss
|
||||
log_prefix: softmax
|
||||
log_accuracy: true
|
||||
|
||||
model_cfg:
|
||||
model: SwinGait
|
||||
Backbone:
|
||||
mode: p3d
|
||||
in_channels: 1
|
||||
layers:
|
||||
- 1
|
||||
- 4
|
||||
- 4
|
||||
- 2
|
||||
channels:
|
||||
- 64
|
||||
- 128
|
||||
bin_num:
|
||||
- 15
|
||||
SeparateBNNecks:
|
||||
in_channels: 256
|
||||
class_num: 3000
|
||||
parts_num: 16
|
||||
|
||||
optimizer_cfg:
|
||||
lr: 0.0003
|
||||
solver: AdamW
|
||||
weight_decay: 0.02
|
||||
|
||||
scheduler_cfg:
|
||||
gamma: 0.1
|
||||
milestones: # Learning Rate Reduction at each milestones
|
||||
- 100000000
|
||||
scheduler: MultiStepLR
|
||||
|
||||
trainer_cfg:
|
||||
enable_float16: true # half_percesion float for memory reduction and speedup
|
||||
fix_BN: false
|
||||
log_iter: 100
|
||||
with_test: false
|
||||
optimizer_reset: True
|
||||
restore_ckpt_strict: false
|
||||
restore_hint: ./output/Gait3D/DeepGaitV2/DeepGaitV2_B1441_C2_P3D/checkpoints/DeepGaitV2_B1441_C2_P3D-60000.pt
|
||||
save_iter: 20000
|
||||
save_name: SwinGait3D_B1442_C2
|
||||
sync_BN: true
|
||||
T_max_iter: 60000
|
||||
total_iter: 80000
|
||||
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_skip_num: 4
|
||||
sample_type: fixed_ordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered
|
||||
type: TripletSampler
|
||||
transform:
|
||||
- type: Compose
|
||||
trf_cfg:
|
||||
- type: RandomPerspective
|
||||
prob: 0.2
|
||||
- type: BaseSilTransform
|
||||
- type: RandomHorizontalFlip
|
||||
prob: 0.2
|
||||
- type: RandomRotate
|
||||
prob: 0.2
|
||||
|
||||
|
||||
Reference in New Issue
Block a user