From fa44b59544fa560bfa0d18e902e98dc682b3d1b3 Mon Sep 17 00:00:00 2001 From: vnoblet Date: Fri, 19 Nov 2021 09:19:39 +0000 Subject: [PATCH] Fix tensor already converted with np.logical --- mojo_val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mojo_val.py b/mojo_val.py index e42390d39126..057d11f3b6e2 100644 --- a/mojo_val.py +++ b/mojo_val.py @@ -202,7 +202,7 @@ def plot_predictions_and_labels(extra_stats, threshold): frame = cv2.imread(str(path), 1) # Draw targets that are missed or not - draw_targets(frame, labelsn.numpy(), np.logical_not(targets_not_matched.numpy()), np.ones(labels.shape[0])) + draw_targets(frame, labelsn.numpy(), np.logical_not(targets_not_matched), np.ones(labels.shape[0])) # Draw wrong and good preds that are above a certain threshold draw_targets(frame, predn[:, :4].numpy(), preds_matched.numpy(), preds_pos, rect=True)