未验证 提交 992cf785 编写于 作者: B Bubbliiiing 提交者: GitHub

Update dataloader.py

上级 aff4aa54
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册