liaoxingyu
16655448c2
onnx/trt support
...
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
fec7abc461
finish v0.2 ddp training
2020-07-06 16:57:43 +08:00
liaoxingyu
2f6d999469
add pytorch to caffe converting
...
Summary: update deployment README and support pytorch to caffe converting for basemodel
2020-06-02 18:11:02 +08:00
liaoxingyu
9bef814155
update visulize logger
2020-05-27 22:53:27 +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
9addfb0ae2
feat: support visualizing label list
...
add features to support label list visualization, which can be used
for label correction or check the hardest sample
2020-05-12 21:35:33 +08:00
liaoxingyu
9b6fda3830
style: remove title in visualization
2020-05-11 14:12:29 +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
651e6ba9c4
feat: support multiprocess predictor
...
add asyncpredictor to support multiprocessing feature extraction with dataloader
2020-05-09 18:23:36 +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
9cf222e093
refactor bn_no_bias
2020-04-08 21:04:09 +08:00
liaoxingyu
c6e0176c53
Upload demo.py and example
2020-04-03 15:07:27 +08:00
