mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity
354 Commits 1 Branch 4 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
liaoxingyu
16655448c2 onnx/trt support 
Summary: change model pretrain mode and support onnx/TensorRT export
 2020-07-29 17:43:39 +08:00
liaoxingyu
3840f3f79a fix arcface NaN problem 
Summary: fix classifier init bugs, which will not initialize classifier weights when use arcface or circle loss.
In this way, it will lead loss NaN problem.
 2020-06-18 12:05:44 +08:00
liaoxingyu
5528d17ace refactor code 
Summary: change code style and refactor code, add avgmax pooling layer in gem_pool
 2020-05-28 13:49:39 +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