20 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
ee634df290 rm _all_ in resnet 2020-07-17 19:40:25 +08:00
liaoxingyu
5ad22d5d36 update regnet init 2020-07-17 19:35:40 +08:00
liaoxingyu
f9539be683 add regnet config file 2020-07-17 19:32:36 +08:00
liaoxingyu
3b57dea49f support regnet backbone 2020-07-17 19:13:45 +08:00
liaoxingyu
f8d468647c resnet expansion add 2020-07-10 22:40:07 +08:00
liaoxingyu
fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu
3732f94405 update osnet 2020-06-09 14:38:49 +08:00
liaoxingyu
a1cb123cfa fix R101 bottleneck missing problem
Summary: add key 101 in block dict to support R101
2020-05-26 14:48:32 +08:00
liaoxingyu
5982f90920 support loading various pretrained weights
Summary: Support loading pretrained model by custom path. With this function, we can load infoMin weights.
2020-05-26 14:33:18 +08:00
liaoxingyu
5d4758125d support ResNet34 backbone
Summary: add BasicBlock to support ResNet34
2020-05-26 13:18:09 +08:00
liaoxingyu
8ab0bc2455 style(backbone): make parameters loading logging more elegant 2020-05-08 12:22:06 +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
e3ae03cc58 feat($modeling/backbones): add new backbones
add osnet, resnext and resnest backbone supported
2020-04-24 12:14:56 +08:00
liaoxingyu
6c9af664dc refactor($modeling/meta_arch): remove useless parts
remove useless meta_archs and backbones
2020-04-21 11:42:14 +08:00
liaoxingyu
95a3c62ad2 refactor(fastreid)
refactor architecture
2020-04-20 10:59:29 +08:00
liaoxingyu
23bedfce12 update version0.2 code 2020-03-25 10:58: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