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

set visrank_topk=10 as default

pull/218/head
KaiyangZhou 2019-08-03 22:59:10 +01:00
parent 28189cf1a6
commit d021fafff1
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/default_parser.py
View File

@ -164,7 +164,7 @@ def init_parser():
parser.add_argument('--visrank', action='store_true',
help='visualize ranked results, only available in evaluation mode')
parser.add_argument('--visrank-topk', type=int, default=15,
parser.add_argument('--visrank-topk', type=int, default=10,
help='visualize topk ranks')
parser.add_argument('--visactmap', action='store_true',
help='visualize CNN activation maps')

3
torchreid/utils/reidtools.py
View File

@ -20,7 +20,7 @@ GREEN = (0, 255, 0)
RED = (0, 0, 255)
def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256, save_dir='', topk=20):
def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256, save_dir='', topk=10):
"""Visualizes ranked results.
Supports both image-reid and video-reid.
@ -37,6 +37,7 @@ def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256,
height (int, optional): resized image height. Default is 256.
save_dir (str): directory to save output images.
topk (int, optional): denoting top-k images in the rank list to be visualized.
Default is 10.
"""
num_q, num_g = distmat.shape
mkdir_if_missing(save_dir)