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

52 Commits (1db050b51f4e722e76e2fcba976255647ae67c0b)

Author SHA1 Message Date
zuchen.wang 6687df06e0 update default config 2021-10-27 15:24:25 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
zuchen.wang 3e753612a9 change pair evaluator 2021-10-18 15:36:54 +08:00
zuchen.wang d162c5fc4e add PairHead 2021-10-15 21:26:16 +08:00
liaoxingyu 6300bd756e Bugfix for cls_layer 
In `any_softmax`, all operations are in-place, so pass into the `logits.clone()` to prevent outside logits changed.
 2021-05-31 17:32:24 +08:00
liaoxingyu 0c8e3d9805 update imbalanced sampler 
Summary: add a new sampler, which is useful for imbalanced or long-tail dataset. This refers to ufoym/imbalanced-dataset-sampler.
 2021-04-21 17:05:10 +08:00
liaoxingyu 44cee30dfc update fastreid v1.2 
Summary:
1. refactor dataloader and heads
2. bugfix in fastattr, fastclas, fastface and partialreid
3. partial-fc supported in fastface
 2021-04-02 21:33:13 +08:00
Xingyu Liao be0a089e1f
bugfix & merge classification transforms (#448) 
Summary: change heads definition in project and config file, merge classification transforms into default transforms
 2021-03-30 15:47:14 +08:00
Xingyu Liao 890224f25c
support classification in fastreid (#443) 
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
 2021-03-26 20:17:39 +08:00
Xingyu Liao 883fd4aede
add configurable decorator & linear loss decouple (#441) 
Summary: Add configurable decorator which can call `Baseline` with `Baseline(cfg)` or `Baseline(cfg, heads=heads, ...)`
Decouple linear and loss computation for partial-fc support.

Reviewed By: l1aoxingyu
 2021-03-23 12:10:06 +08:00
liaoxingyu b5c3c0a24d update docs 2021-01-22 21:11:19 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu a327a70f0d v0.3 update 
Summary:
1. change DPP training in apex way;
2. make warmup scheduler by iter and lr scheduler by epoch;
3. replace random erasing with torchvision implementation;
4. naming modification in config file
 2020-12-07 14:19:20 +08:00
liaoxingyu 5dfe537515 update attribute project 2020-09-23 19:45:13 +08:00
liaoxingyu d9a63a959f fix bug in mgn #272 
Summary: fix get_norm bug in mgn
 2020-09-17 18:17:53 +08:00
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 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