mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix visualization problem #233

pull/240/head
liaoxingyu 2020-08-20 17:03:43 +08:00
parent 07872d4cdb
commit 8d4bd77ae3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/predictor.py
View File

@ -123,7 +123,7 @@ class AsyncPredictor:
for gpuid in range(max(num_gpus, 1)): for gpuid in range(max(num_gpus, 1)):
cfg = cfg.clone() cfg = cfg.clone()
cfg.defrost() cfg.defrost()
cfg.MODEL.DEVICE = "cuda: {}".format(gpuid) if num_gpus > 0 else "cpu" cfg.MODEL.DEVICE = "cuda:{}".format(gpuid) if num_gpus > 0 else "cpu"
self.procs.append( self.procs.append(
AsyncPredictor._PredictWorker(cfg, self.task_queue, self.result_queue) AsyncPredictor._PredictWorker(cfg, self.task_queue, self.result_queue)
) )

2
demo/visualize_result.py
View File

@ -120,7 +120,7 @@ if __name__ == '__main__':
distmat = distmat.numpy() distmat = distmat.numpy()
logger.info("Computing APs for all query images ...") logger.info("Computing APs for all query images ...")
cmc, all_ap, all_inp = evaluate_rank(distmat, q_pids, g_pids, q_camids, g_camids) cmc, all_ap, all_inp = evaluate_rank(distmat, q_feat, g_feat, q_pids, g_pids, q_camids, g_camids)
visualizer = Visualizer(test_loader.dataset) visualizer = Visualizer(test_loader.dataset)
visualizer.get_model_output(all_ap, distmat, q_pids, g_pids, q_camids, g_camids) visualizer.get_model_output(all_ap, distmat, q_pids, g_pids, q_camids, g_camids)