From 2cac19ce31a0a650f58982f25841ed1af1676f1a Mon Sep 17 00:00:00 2001
From: ViokingTung <18394173714@163.com>
Date: Thu, 2 Sep 2021 14:30:18 +0800
Subject: [PATCH] fix num_vis type error bug in visualizer.py (#566)
---
fastreid/utils/visualizer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/fastreid/utils/visualizer.py b/fastreid/utils/visualizer.py
index 5a06abd..432d6e0 100644
--- a/fastreid/utils/visualizer.py
+++ b/fastreid/utils/visualizer.py
@@ -153,7 +153,7 @@ class Visualizer:
query_indices = np.argsort(self.all_ap)
if rank_sort == 'descending': query_indices = query_indices[::-1]
- query_indices = query_indices[:num_vis]
+ query_indices = query_indices[:int(num_vis)]
self.save_rank_result(query_indices, output, max_rank, vis_label, label_sort, actmap)
def vis_roc_curve(self, output):