liaoxingyu
a327a70f0d
v0.3 update
...
Summary:
1. change DPP training in apex way;
2. make warmup scheduler by iter and lr scheduler by epoch;
3. replace random erasing with torchvision implementation;
4. naming modification in config file
2020-12-07 14:19:20 +08:00
liaoxingyu
10cbaab155
support finetuning from trained models
...
Summary: add a flag for supporting finetuning model from the trained weights, and it's very useful when performing across domain reid
2020-09-28 17:10:10 +08:00
liaoxingyu
3024cea3a3
add more datasets
...
Summary: add wildtracker.py datasets and cuhk_sysu.py datasets.
2020-09-28 17:05:40 +08:00
liaoxingyu
154a06b875
refactor code
2020-09-23 19:31:46 +08:00
liaoxingyu
919a515eb7
fix combine all training
2020-09-09 15:28:21 +08:00
liaoxingyu
866a196d19
add more datasets support
2020-09-01 16:13:12 +08:00
liaoxingyu
9c667d1a0f
add pretrain_path support in backbone
2020-08-14 14:00:26 +08:00
liaoxingyu
d1c20cbe50
fix pre-train model bugs
...
fix bugs locks when downloading pre-train model
2020-08-04 15:56:36 +08:00
liaoxingyu
977aa4a357
update model url
2020-07-31 14:59:37 +08:00
liaoxingyu
65169b40bd
support ddp testing
2020-07-30 20:15:28 +08:00
liaoxingyu
16655448c2
onnx/trt support
...
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
c851c7edc8
fix dataset prefix #76
2020-07-06 16:54:54 +08:00
liaoxingyu
8879db3fba
update training instruction
...
Summary: update dataset configuration and training instruction
2020-06-16 19:43:36 +08:00
liaoxingyu
fd90555e19
feat: add multi-dataset joint training
...
new feature that can support joint training, and find some bugs in funtion combine_all of datasets/bases.py that assume person id in dataset has been relabeld from 0 to num_class.
Another bug appears in msmt17 which trainset and testset person id both begin from 0, and we should change testset id from num_class of trainset.
2020-05-18 20:06:04 +08:00
Jinkai Zheng
640c9bfc97
Add small, medium and large vehicle test datasets
2020-05-11 01:34:19 -05:00
Jinkai Zheng
e059b751a6
Add modified vehicle datasets with small, medium and large test datastes
2020-05-11 01:30:56 -05:00
zjk15068083791
2f3f6e3267
Add files via upload
2020-04-27 16:27:44 +08:00
liaoxingyu
bb50b6c5a7
docs($projects): update agw readme
2020-04-21 11:35:54 +08:00
liaoxingyu
9684500a57
chagne arch
...
1. change dataset show to trainset show and testset show seperately
2. add cls layer to easily plug in circle loss and arcface
2020-04-19 12:54:01 +08:00
liaoxingyu
be9faa5605
update focal loss
...
update dataset info display
update seperate lr
update adaptive label smooth regularization
2020-04-17 13:46:10 +08:00
liaoxingyu
6a8961ce48
1. upload circle loss and arcface
...
2. finish freeze training
3. update augmix data augmentation
2020-04-05 23:54:26 +08:00
L1aoXingyu
8a9c0ccfad
Finish first version for fastreid
2020-02-10 22:13:04 +08:00
L1aoXingyu
db6ed12b14
Update sampler code
2020-02-10 07:38:56 +08:00
liaoxingyu
71950d2c09
1. Fix evaluation code
...
2. Finish multi-dataset evaluation
3. Decouple image preprocess and output postprocess with model forward for DataParallel training
4. Finish build backbone registry
5. Fix dataset sampler
2020-01-21 20:24:26 +08:00
liaoxingyu
b761b656f3
Finish basic training loop and evaluation results
2020-01-20 21:33:37 +08:00
