liaoxingyu
9fae467adf
feat(engine/defaults): add DefaultPredictor to get image reid features
...
Add a new predictor interface, and modify demo code to predict image features.
2020-05-08 19:24:27 +08:00
liaoxingyu
0b15ac4e03
feat(hooks&optim): update stochastic weight averging hooks
...
Update swa method which will do after regular training if you
set this option enabled.
2020-05-08 12:20:04 +08:00
liaoxingyu
afac8aad5d
Fix(engine): fix preciseBN dataloader bugs
...
preciseBN needs to pass data prefetcher, but now a DataLoader is passed
2020-05-06 14:26:34 +08:00
liaoxingyu
6d96529d4c
fix(data): fix resume training bug
...
fix dataset pid dictionary loading bug when resume training,
data prefetcher will pre-load a batch of data, this will lead to
misalignment of old pid dict and updated pid dict.
We can address this problem by redefine a prefetcher in resume_or_load
2020-05-05 23:20:42 +08:00
liaoxingyu
a2dcd7b4ab
feat(layers/norm): add ghost batchnorm
...
add a get_norm fucntion to easily change normalization between batchnorm, ghost bn and group bn
2020-05-01 09:02:46 +08:00
liaoxingyu
d27729c5bb
refactor(preciseBN): add preciseBN datasets show
2020-04-29 21:05:53 +08:00
liaoxingyu
e38a799b63
fix(engine/defaults): fix precise bn bug
...
fix problem in precise bn, which will not use precise bn datasets, and throw some errors
2020-04-29 16:16:54 +08:00
liaoxingyu
4d2fa28dbb
update freeze layer
...
update preciseBN
update circle loss with metric learning and cross entropy loss form
update loss call methods
2020-04-06 23:34:27 +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
liaoxingyu
23bedfce12
update version0.2 code
2020-03-25 10:58:26 +08:00
L1aoXingyu
b020c7f0ae
Fix data prefetcher minor bug
2020-02-27 12:16:57 +08:00
L1aoXingyu
12957f66aa
Change architecture:
...
1. delete redundant preprocess
2. add data prefetcher to accelerate data loading
3. fix minor bug of triplet sampler when only one image for one id
2020-02-18 21:01:23 +08:00
L1aoXingyu
327d74ffbb
Update strong baseline result
...
Change data sampler
2020-02-13 00:19:15 +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
