mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix bug: add 'args.pool' to test()

pull/62/head
KaiyangZhou 2018-08-28 09:57:12 +01:00
parent cc1e23049a
commit 0084a30c6a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_vidreid_xent.py
View File

@ -197,7 +197,7 @@ def main():
if args.evaluate:
print("Evaluate only")
distmat = test(model, queryloader, galleryloader, use_gpu, return_distmat=True)
distmat = test(model, queryloader, galleryloader, args.pool, use_gpu, return_distmat=True)
if args.vis_ranked_res:
visualize_ranked_results(
distmat, dataset,

2
train_vidreid_xent_htri.py
View File

@ -198,7 +198,7 @@ def main():
if args.evaluate:
print("Evaluate only")
distmat = test(model, queryloader, galleryloader, use_gpu, return_distmat=True)
distmat = test(model, queryloader, galleryloader, args.pool, use_gpu, return_distmat=True)
if args.vis_ranked_res:
visualize_ranked_results(
distmat, dataset,