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
