crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.0.0 official release (#18)

Browse Source 
* fix bug in fix_BN

* gaitgl OUMVLP support.

* update ./doc/3.advance_usage.md Cross-Dataset Evalution & Data Agumentation

* update config

* update docs.3

* update docs.3

* add loss doc and gather input decorator

* refine the create model doc

* support rearrange directory of unzipped OUMVLP

* fix some bugs in loss_aggregator.py

* refine docs and little fix

* add oumvlp pretreatment description

* pretreatment dataset fix oumvlp description

* add gaitgl oumvlp result

* assert gaitgl input size

* add pipeline

* update the readme.

* update pipeline and readme

* Corrigendum.

* add logo and remove path

* update new logo

* Update README.md

* modify logo size

Co-authored-by: 12131100 <12131100@mail.sustech.edu.cn>
Co-authored-by: noahshen98 <77523610+noahshen98@users.noreply.github.com>
Co-authored-by: Noah <595311942@qq.com>
This commit is contained in:
Junhao Liang
2021-12-08 20:05:28 +08:00
committed by GitHub
parent 6e71c7ac34
commit bb6cd5149a
39 changed files with 11401 additions and 230 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/baseline.yaml
+8 -8
View File
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: CASIA-B
dataset_root: your_path
dataset_root: your_path
dataset_partition: ./misc/partitions/CASIA-B_include_005.json
num_workers: 1
remove_no_gallery: false # Remove probe if no gallery for it
@@ -18,9 +18,9 @@ evaluator_cfg:
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
transform:
- type: BaseSilCuttingTransform
img_w: 64
loss_cfg:
- loss_term_weights: 1.0
@@ -79,7 +79,7 @@ scheduler_cfg:
scheduler: MultiStepLR
trainer_cfg:
enable_float16: true # half_percesion float for memory reduction and speedup
fix_BN: false
fix_BN: true
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
@@ -97,6 +97,6 @@ trainer_cfg:
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
transform:
- type: BaseSilCuttingTransform
img_w: 64
config/baseline_OUMVLP.yaml
+102
View File
@@ -0,0 +1,102 @@
data_cfg:
dataset_name: OUMVLP
dataset_root: your_path
dataset_partition: ./misc/partitions/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
eval_func: identification
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_weights: 1.0
margin: 0.2
type: TripletLoss
log_prefix: triplet
- loss_term_weights: 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
config/gaitgl.yaml
-1
View File
@@ -49,7 +49,6 @@ scheduler_cfg:
trainer_cfg:
enable_distributed: true
enable_float16: false
fix_BN: false
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
config/gaitgl_OUMVLP.yaml
+69
View File
@@ -0,0 +1,69 @@
# Note : *** the batch_size should be equal to the gpus number at the test phase!!! ***
data_cfg:
dataset_name: OUMVLP
dataset_root: your_path
dataset_partition: ./misc/partitions/OUMVLP.json
num_workers: 1
remove_no_gallery: false
test_dataset_name: OUMVLP
evaluator_cfg:
enable_distributed: true
enable_float16: false
restore_ckpt_strict: true
restore_hint: 210000
save_name: GaitGL
sampler:
batch_size: 2
sample_type: all_ordered
type: InferenceSampler
loss_cfg:
- loss_term_weights: 1.0
margin: 0.2
type: TripletLoss
log_prefix: triplet
- loss_term_weights: 1.0
scale: 1
type: CrossEntropyLoss
log_accuracy: true
label_smooth: true
log_prefix: softmax
model_cfg:
model: GaitGL
channels: [32, 64, 128, 256]
class_num: 5153
optimizer_cfg:
lr: 1.0e-4
solver: Adam
weight_decay: 0
scheduler_cfg:
gamma: 0.1
milestones:
- 150000
- 200000
scheduler: MultiStepLR
trainer_cfg:
enable_distributed: true
enable_float16: true
with_test: false
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
save_iter: 10000
save_name: GaitGL
sync_BN: true
total_iter: 210000
sampler:
batch_shuffle: true
batch_size:
- 32
- 8
frames_num_fixed: 30
frames_skip_num: 0
sample_type: fixed_ordered
type: TripletSampler
config/gaitpart.yaml
-1
View File
@@ -58,7 +58,6 @@ scheduler_cfg:
trainer_cfg:
enable_float16: true
fix_BN: false
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
config/gaitpart_OUMVLP.yaml
+82
View File
@@ -0,0 +1,82 @@
data_cfg:
dataset_name: OUMVLP
dataset_root: your_path
dataset_partition: ./misc/partitions/OUMVLP.json
num_workers: 4
remove_no_gallery: false
test_dataset_name: OUMVLP
evaluator_cfg:
enable_float16: false
restore_ckpt_strict: true
restore_hint: 250000
save_name: GaitPart
sampler:
batch_size: 4
sample_type: all_ordered
type: InferenceSampler
metric: euc # cos
loss_cfg:
loss_term_weights: 1.0
margin: 0.2
type: TripletLoss
log_prefix: triplet
model_cfg:
model: GaitPart
backbone_cfg:
in_channels: 1
layers_cfg:
- BC-32
- BC-32
- M
- BC-64
- BC-64
- M
- FC-128-3
- FC-128-3
- FC-256-3
- FC-256-3
type: Plain
SeparateFCs:
in_channels: 256
out_channels: 256
parts_num: 16
bin_num:
- 16
optimizer_cfg:
lr: 0.0001
momentum: 0.9
solver: Adam
weight_decay: 0.0
scheduler_cfg:
gamma: 0.1
milestones:
- 150000
scheduler: MultiStepLR
trainer_cfg:
enable_float16: true
fix_BN: false
log_iter: 100
with_test: true
restore_ckpt_strict: true
restore_hint: 0
save_iter: 10000
save_name: GaitPart
sync_BN: false
total_iter: 250000
sampler:
batch_shuffle: false
batch_size:
- 32
- 16
frames_num_fixed: 30
frames_num_max: 50
frames_num_min: 25
frames_skip_num: 10
sample_type: fixed_ordered
type: TripletSampler
config/gaitset.yaml
-1
View File
@@ -56,7 +56,6 @@ scheduler_cfg:
trainer_cfg:
enable_float16: true
fix_BN: false
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
config/gaitset_OUMVLP.yaml
+76
View File
@@ -0,0 +1,76 @@
data_cfg:
dataset_name: OUMVLP
dataset_root: your_path
dataset_partition: ./misc/partitions/OUMVLP.json
num_workers: 4
remove_no_gallery: false
test_dataset_name: OUMVLP
evaluator_cfg:
enable_float16: true
restore_ckpt_strict: true
restore_hint: 250000
save_name: GaitSet
sampler:
batch_size: 4
sample_type: all_ordered
type: InferenceSampler
metric: euc # cos
loss_cfg:
loss_term_weights: 1.0
margin: 0.2
type: TripletLoss
log_prefix: triplet
model_cfg:
model: GaitSet
in_channels:
- 1
- 64
- 128
- 256
SeparateFCs:
in_channels: 256
out_channels: 256
parts_num: 62
bin_num:
- 16
- 8
- 4
- 2
- 1
optimizer_cfg:
lr: 0.0001
momentum: 0.9
solver: Adam
weight_decay: 0
scheduler_cfg:
gamma: 0.1
milestones:
- 150000
scheduler: MultiStepLR
trainer_cfg:
enable_float16: true
fix_BN: false
with_test: true
log_iter: 100
restore_ckpt_strict: true
restore_hint: 0
save_iter: 10000
save_name: GaitSet
sync_BN: false
total_iter: 250000
sampler:
batch_shuffle: false
batch_size:
- 32
- 16
frames_num_fixed: 30
frames_num_max: 50
frames_num_min: 25
sample_type: fixed_unordered
type: TripletSampler
config/gln/gln_phase1.yaml
+1 -1
View File
@@ -76,7 +76,7 @@ scheduler_cfg:
trainer_cfg:
enable_distributed: true
enable_float16: true
fix_BN: false
fix_layers: false
with_test: false
log_iter: 100
optimizer_reset: false
config/gln/gln_phase2.yaml
+1 -1
View File
@@ -71,7 +71,7 @@ scheduler_cfg:
trainer_cfg:
enable_distributed: true
enable_float16: true
fix_BN: false
fix_layers: false
log_iter: 100
optimizer_reset: true
scheduler_reset: true