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