Commit Graph

23 Commits (master)

Author SHA1 Message Date
Xingyu Liao 4508251d74 update docs 2022-01-24 12:56:52 +08:00
liaoxingyu 91ff631184 Minor changes
Some minor changes, such as class name changing, remove extra blank line, etc.
2021-05-31 17:27:14 +08:00
liaoxingyu dbf1604231 fix model deploy problems 2021-05-25 15:55:43 +08:00
Xingyu Liao bb6ddbf8b1
bugfix for visualize and demo (#468)
Summary minor bugs for visualize and demo caused by revised DataLoader
2021-04-21 16:24:34 +08:00
liaoxingyu 1b9799f601 fix demo eval bug (#345)
Summary: Fix bug in demo.py when computing cmc and mAP in `evaluate_rank`. This is because after changing eval function, the passing params in demo are not changed.

close #345
2020-12-04 11:14:18 +08:00
liaoxingyu 1f4e4909eb fix typro ##255
fix #255
2020-09-10 11:10:40 +08:00
liaoxingyu 9fb3285d5c rm extra sh file 2020-08-21 18:00:36 +08:00
liaoxingyu 8d4bd77ae3 fix visualization problem #233 2020-08-20 17:03:43 +08:00
liaoxingyu 64861ea6c8 update config file 2020-07-29 19:48:49 +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 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