From 992cf785319fffd00e144e61f8aba85700633bbc Mon Sep 17 00:00:00 2001 From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com> Date: Tue, 8 Sep 2020 13:21:50 +0800 Subject: [PATCH] Update dataloader.py --- utils/dataloader.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/dataloader.py b/utils/dataloader.py index ed6a1b3..7c4950d 100644 --- a/utils/dataloader.py +++ b/utils/dataloader.py @@ -250,6 +250,5 @@ def yolo_dataset_collate(batch): images.append(img) bboxes.append(box) images = np.array(images) - bboxes = np.array(bboxes) return images, bboxes -- GitLab