From 7c518680864d8c9320333de7f913d71c43dc0954 Mon Sep 17 00:00:00 2001
From: Manusha Chethiyawardhana <45126869+Manusha17@users.noreply.github.com>
Date: Thu, 5 Aug 2021 20:57:27 +0530
Subject: [PATCH] Bug fix in camera.py

---
 tool/camera.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tool/camera.py b/tool/camera.py
index d3692b68..7f7a3918 100644
--- a/tool/camera.py
+++ b/tool/camera.py
@@ -65,7 +65,7 @@ def arg_parse():
             sized = cv2.cvtColor(sized, cv2.COLOR_BGR2RGB)
             boxes = do_detect(model, sized, 0.5, 0.4, CUDA)
 
-            orig_im = plot_boxes_cv2(frame, boxes, class_names=class_names)
+            orig_im = plot_boxes_cv2(frame, boxes[0], class_names=class_names)
 
             cv2.imshow("frame", orig_im)
             key = cv2.waitKey(1)