updates

2019-04-26 14:17:04 +02:00 · 2019-04-26 14:17:04 +02:00 · 0bed0a9a0e
commit 0bed0a9a0e
parent 3b134e3a84
1 changed files with 2 additions and 2 deletions
--- 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):