8 Commits (9b5af4166e772f00ddc10f26fbb1f6b6546ad19a)

Author	SHA1	Message	Date
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	8d4bd77ae3	fix visualization problem #233	2020-08-20 17:03:43 +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	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