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

42 Commits (54f96ba78ad087c7110bafe2132a91536f0b5a01)

Author SHA1 Message Date
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
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
liaoxingyu f57c5764e3 support multi-node training 2021-03-09 20:07:28 +08:00
liaoxingyu 68c190b53c replace list in evaluator process with dict 2021-03-09 20:07:13 +08:00
liaoxingyu a53fd17874 update docs 2021-01-23 15:25:58 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu 66941cf27a feat: support flip testing 2020-12-22 15:50:50 +08:00
liaoxingyu 5469e8ce76 feat: add save best model mechanism 2020-12-22 15:49:46 +08:00
liaoxingyu2 64bf78afee update rank_cylib 2020-11-06 11:00:13 +08:00
liaoxingyu2 3bd2fad9a5 support faiss-based rerank 
Summary: accelerate rerank with faiss-gpu
 2020-11-06 10:59:53 +08:00
liaoxingyu 5b88736e1d refactor evaluation code 2020-09-23 19:32:40 +08:00
liaoxingyu d00ce8fc3c refactor model arch 2020-09-01 16:14:45 +08:00
liaoxingyu 9c667d1a0f add pretrain_path support in backbone 2020-08-14 14:00:26 +08:00
liaoxingyu ae7c9288cf support faiss retrieval and cython roc evaluation 2020-08-12 16:27:57 +08:00
Xingyu Liao f74cebcd88
fix rank_cylib bug about valid query index 
fix bugs of rank_cylib

Reviewed by: l1aoxingyu
 2020-08-10 19:28:05 +08:00
liaoxingyu 35794621cc remove addmm warning 2020-07-31 16:32:10 +08:00
liaoxingyu 65169b40bd support ddp testing 2020-07-30 20:15:28 +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 eb88076714 update tabulate vis 2020-07-15 17:12:24 +08:00
liaoxingyu cf06c9e8c7 update tabulate vis with dataset name 2020-07-15 15:53:03 +08:00
liaoxingyu e0fc15269c update tabulate vis 2020-07-15 14:56:18 +08:00
liaoxingyu ea8a3cc534 fix typro 2020-07-10 16:26:35 +08:00
grimoire 76afdd8d66 fix bugs of rank_cylib 2020-07-07 11:12:46 +08:00
liaoxingyu fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu 84c733fa85 fix: remove prefetcher, put normalizer in model 
1. remove messy data prefetcher which will cause  confusion
2. put normliazer in model to accelerate training via GPU computing
 2020-05-25 23:39:11 +08:00
liaoxingyu 2ac55a7601 feat: update roc curve and TPR@FPR metric 
support plot multiple ROC curves with different model
 2020-05-20 14:29:33 +08:00
liaoxingyu e344eae1cc feat: support plotting roc curve and compute auc score 
ROC curve and AUC score will be help for thresholds
 2020-05-19 20:45:26 +08:00
liaoxingyu 579a5cf552 fix: modify re-rank typro 2020-05-18 17:05:20 +08:00
liaoxingyu 5ae3d4fecf feat: add aqe support in test phase 
query expansion will combine the retrived topk nearest neighbors with the original query feature,
it will enhance mAP by a large margin.feat:
 2020-05-13 16:27:22 +08:00
liaoxingyu 320010f2ae feat: support re-rank in test phase 2020-05-13 11:47:52 +08:00
liaoxingyu 13bb03eb07 feat: add rank result visualization tools 
Update visualization tools which can save rank list with AP metrics from high to low, vice versa.
In order to compute AP fast in visualizer, modify rank_cylib to get all_AP instead of mAP.
In this way, we can use Cython to compute results.
 2020-05-10 23:17:10 +08:00
liaoxingyu 9fae467adf feat(engine/defaults): add DefaultPredictor to get image reid features 
Add a new predictor interface, and modify demo code to predict image features.
 2020-05-08 19:24:27 +08:00
liaoxingyu 4d3e5fd378 refactor(evaluation): add feature l2 norm in evaluation 
change the l2 norm function from inference function in Module to reid evaluation.
because sometimes we need to use the original features generated by model rather than normalized ones.
 2020-04-27 14:51:39 +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 23bedfce12 update version0.2 code 2020-03-25 10:58:26 +08:00
L1aoXingyu b020c7f0ae Fix data prefetcher minor bug 2020-02-27 12:16:57 +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 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