From 2913b23fb1f3409bdc1086069bf304990c67d07f Mon Sep 17 00:00:00 2001 From: Nikita Kotov Date: Wed, 15 Mar 2023 21:22:04 +0300 Subject: [PATCH] fixed function return type error 'output_to_target' --- utils/plots.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/plots.py b/utils/plots.py index c90a96b8..8f33b9c7 100644 --- a/utils/plots.py +++ b/utils/plots.py @@ -105,7 +105,7 @@ def output_to_target(output, width, height): targets.append([i, cls, x, y, w, h, conf]) - return np.array(targets) + return torch.tensor(targets) def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max_size=640, max_subplots=16):