liaoxingyu
|
ac8409a7da
|
updating for pytorch1.6
|
2020-08-20 15:51:41 +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
|
3b57dea49f
|
support regnet backbone
|
2020-07-17 19:13:45 +08:00 |
liaoxingyu
|
3f35eb449d
|
minor update
|
2020-07-14 11:58:06 +08:00 |
liaoxingyu
|
5ae2cff47e
|
fix circle/arcface pred_logits
fix #136
|
2020-07-06 16:57:03 +08:00 |
liaoxingyu
|
56a1ab4a5d
|
update fast global avgpool
Summary: update fast pool according to https://arxiv.org/pdf/2003.13630.pdf
|
2020-06-12 16:34:03 +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
|
5d4758125d
|
support ResNet34 backbone
Summary: add BasicBlock to support ResNet34
|
2020-05-26 13:18:09 +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
|
329764bb60
|
refactor(heads): move num_classes out from heads
set parameter num_classes in meta_arch to easily modify different heads fc layer
|
2020-04-29 21:29:48 +08:00 |
liaoxingyu
|
9910bb9158
|
fix($modeling/heads): fix targets missing bug
fix bug in heads about return outputs without targets.
|
2020-04-27 14:49:58 +08:00 |
liaoxingyu
|
95a3c62ad2
|
refactor(fastreid)
refactor architecture
|
2020-04-20 10:59:29 +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
|
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
|
23bedfce12
|
update version0.2 code
|
2020-03-25 10:58:26 +08:00 |