From 0bed0a9a0e0c56925d7fd7222c41175647a9918e Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 26 Apr 2019 14:17:04 +0200
Subject: [PATCH] updates

---
 test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test.py b/test.py
index b260714e..c9516953 100644
--- a/test.py
+++ b/test.py
@@ -113,8 +113,8 @@ def test(
 
                 # target boxes
                 tbox = xywh2xyxy(labels[:, 1:5])
-                tbox[[0, 2]] *= width
-                tbox[[1, 3]] *= height
+                tbox[:, [0, 2]] *= width
+                tbox[:, [1, 3]] *= height
 
                 # Search for correct predictions
                 for i, (*pbox, pconf, pcls_conf, pcls) in enumerate(pred):