From 47c1fcaf08747e473ff0d5689476e091a20d0427 Mon Sep 17 00:00:00 2001 From: Kevin Huang Date: Wed, 30 Jun 2021 18:41:51 -0700 Subject: [PATCH] fix decode netout row --- yolo3_one_file_to_detect_them_all.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yolo3_one_file_to_detect_them_all.py b/yolo3_one_file_to_detect_them_all.py index 17ca64e22..28ff080aa 100644 --- a/yolo3_one_file_to_detect_them_all.py +++ b/yolo3_one_file_to_detect_them_all.py @@ -291,7 +291,7 @@ def decode_netout(netout, anchors, obj_thresh, nms_thresh, net_h, net_w): netout[..., 5:] *= netout[..., 5:] > obj_thresh for i in range(grid_h*grid_w): - row = i / grid_w + row = i // grid_w col = i % grid_w for b in range(nb_box):