This is an automated email from the ASF dual-hosted git repository.
anijain2305 pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 7388715 Fix _get_yolo_detections (#8477)
7388715 is described below
commit 73887156321fcee1700ef8661f052d8d38022a4d
Author: Alexander Pivovarov <pivo...@amazon.com>
AuthorDate: Fri Jul 16 10:26:25 2021 -0700
Fix _get_yolo_detections (#8477)
---
python/tvm/relay/testing/yolo_detection.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/relay/testing/yolo_detection.py
b/python/tvm/relay/testing/yolo_detection.py
index a387f30..949d024 100644
--- a/python/tvm/relay/testing/yolo_detection.py
+++ b/python/tvm/relay/testing/yolo_detection.py
@@ -103,8 +103,8 @@ def _get_yolo_detections(l, im_shape, net_shape, thresh,
relative, dets):
l["biases"],
np.asarray(l["mask"])[location[0]],
location,
- data.shape[2],
data.shape[3],
+ data.shape[2],
net_shape[0],
net_shape[1],
)
@@ -139,10 +139,10 @@ def _get_region_detections(l, im_shape, net_shape,
thresh, relative, dets):
l["biases"],
n,
location,
- data.shape[2],
data.shape[3],
data.shape[2],
data.shape[3],
+ data.shape[2],
)
objectness = scale if scale > thresh else 0
if objectness:
