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

36 Commits (77caa01e345ee0f340b60d7d3320c9803c719ca4)

Author SHA1 Message Date
liaoxingyu d00ce8fc3c refactor model arch 2020-09-01 16:14:45 +08:00
liaoxingyu f4305b0964 fix bnneck 2020-08-20 16:21:14 +08:00
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 ea8a3cc534 fix typro 2020-07-10 16:26:35 +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 c036ac5bdd update reduction head 2020-05-30 16:50:02 +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 94c86579a3 fix(heads): fix bug in reduce head 
add neck_feat from config, add inplace in leakyrelu for memory save
 2020-05-23 10:41:13 +08:00
liaoxingyu c21de64166 fix: add linear initial method 2020-05-21 23:59:51 +08:00
liaoxingyu 0356ef8c5c feat: add SyncBN and GroupNorm suppor 2020-05-14 11:36:28 +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 2efbc6d371 fix($modeling/heads/bnneck_head): fix heads outputs bug 
fix bug of heads outputs, which will lead to no targets return.
 2020-04-27 11:48:21 +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 be9faa5605 update focal loss 
update dataset info display
update seperate lr
update adaptive label smooth regularization
 2020-04-17 13:46:10 +08:00
liaoxingyu 9cf222e093 refactor bn_no_bias 2020-04-08 21:04:09 +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 c6e0176c53 Upload demo.py and example 2020-04-03 15:07:27 +08:00
liaoxingyu 91dc9bc71f Merge branch 'master' of github.com:L1aoXingyu/fast-reid 
 Conflicts:
	fastreid/config/defaults.py
	fastreid/layers/gem_pool.py
	fastreid/modeling/backbones/resnet.py
	fastreid/modeling/heads/__init__.py
	fastreid/modeling/heads/build.py
	fastreid/modeling/losses/build.py
	fastreid/modeling/meta_arch/__init__.py
	fastreid/modeling/meta_arch/abd_network.py
	fastreid/modeling/meta_arch/baseline.py
	fastreid/modeling/meta_arch/bdb_network.py
	fastreid/modeling/meta_arch/mf_network.py
	projects/StrongBaseline/configs/Base-Strongbaseline.yml
	projects/StrongBaseline/configs/baseline_dukemtmc.yml
	projects/StrongBaseline/train_net.py
 2020-03-25 11:05:28 +08:00
liaoxingyu 23bedfce12 update version0.2 code 2020-03-25 10:58:26 +08:00
L1aoXingyu b1058118ca update BDB-net code 
update MF-net code
 2020-03-19 12:23:41 +08:00
L1aoXingyu acf363c181 1. Change loss function as a build-in attributes of heads 
2. Update agw and bagtricks result
 2020-03-16 15:23:09 +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 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