utils_vis.py

Go to the documentation of this file.

from PIL import Image
from matplotlib import pyplot as plt
from tqdm import tqdm
import os
 
 
def display_reid_result(sorted_keep_index, crop_image_path, test_image_list):
    columns = 5
    cnt = 0
    rows = (len(sorted_keep_index) // columns)+1
    
    figsize = (30.0, rows*2)
    plt.figure(figsize=figsize)
 
    for index in tqdm(sorted_keep_index):
 
        image_path = os.path.join(crop_image_path, test_image_list[index[0]])
        image = Image.open(image_path)
        image_index = cnt+1
 
        title_txt = f'{test_image_list[index[0]]} | {index[1]:.4f}'
        plt.subplot(rows, columns, image_index)
        plt.title(title_txt)
        plt.xticks([])
        plt.yticks([])
        plt.imshow(image)
        
        cnt+=1
 
    plt.show()