mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
122 Commits
1 Branch
4 Tags
15 MiB
Commit Graph

21 Commits (9addfb0ae2d5a2c0970fe7749167add12df09363)

Author SHA1 Message Date
liaoxingyu 651e6ba9c4 feat: support multiprocess predictor 
add asyncpredictor to support multiprocessing feature extraction with dataloader
 2020-05-09 18:23:36 +08:00
liaoxingyu 4be4cacb73 fix: add a simple way to reset data prefetcher when resume training 
use data prefetcher build-in reset function to reload it rather than
redefining a new data prefetcher, otherwise it will introduce other
problems in eval-only mode.
 2020-05-09 11:58:27 +08:00
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 948af64fd1 feat: add swa algorithm 
Add swa and related config options,
if it is enabled, model will do swa after regular training
 2020-05-06 10:17:44 +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 a2f69d0537 Update StrongBaseline results for market1501 and dukemtmc 2020-02-11 22:38:40 +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