Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-26 14:17:04 +02:00
parent 3b134e3a84
commit 0bed0a9a0e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -113,8 +113,8 @@ def test(
# target boxes # target boxes
tbox = xywh2xyxy(labels[:, 1:5]) tbox = xywh2xyxy(labels[:, 1:5])
tbox[[0, 2]] *= width tbox[:, [0, 2]] *= width
tbox[[1, 3]] *= height tbox[:, [1, 3]] *= height
# Search for correct predictions # Search for correct predictions
for i, (*pbox, pconf, pcls_conf, pcls) in enumerate(pred): for i, (*pbox, pconf, pcls_conf, pcls) in enumerate(pred):