fix num_vis type error bug in visualizer.py (#566)

2021-09-02 14:30:18 +08:00 · 2021-09-02 14:30:18 +08:00 · 2cac19ce31
parent ced654431b
commit 2cac19ce31
1 changed files with 1 additions and 1 deletions
--- 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):