liaoxingyu
|
4d573b8107
|
refactor reid head
Summary: merge BNneckHead, LinearHead and ReductionHead into EmbeddingHead
because they are highly similar and can be prepared for ClsHead
|
2020-09-10 10:57:37 +08:00 |
liaoxingyu
|
d00ce8fc3c
|
refactor model arch
|
2020-09-01 16:14:45 +08:00 |
liaoxingyu
|
ea8a3cc534
|
fix typro
|
2020-07-10 16:26:35 +08:00 |
liaoxingyu
|
fec7abc461
|
finish v0.2 ddp training
|
2020-07-06 16:57:43 +08:00 |
liaoxingyu
|
84c733fa85
|
fix: remove prefetcher, put normalizer in model
1. remove messy data prefetcher which will cause confusion
2. put normliazer in model to accelerate training via GPU computing
|
2020-05-25 23:39:11 +08:00 |
liaoxingyu
|
e990cf3e34
|
style: fix some typro
|
2020-05-21 15:55:51 +08:00 |
liaoxingyu
|
bf18479541
|
fix: revise syncBN bug
|
2020-05-14 14:52:37 +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
|
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
|
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
|
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 |
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
|
a2f69d0537
|
Update StrongBaseline results for market1501 and dukemtmc
|
2020-02-11 22:38:40 +08:00 |
L1aoXingyu
|
db6ed12b14
|
Update sampler code
|
2020-02-10 07:38:56 +08:00 |