Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Comment fix (#3058)

Browse Source 
Co-authored-by: kepler62f <>
This commit is contained in:
kepler62fandkepler62f authored and GitHub committed 2021-05-07 12:28:55 +02:00
1 parent 3ef3a95cfa
commit f2de1ad2aa
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
test.py
+2 -2
View File
@@ -188,8 +188,8 @@ def test(data,
# Per target class # Per target class
for cls in torch.unique(tcls_tensor): for cls in torch.unique(tcls_tensor):
ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1) # prediction indices ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1) # target indices
pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1) # target indices pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1) # prediction indices
# Search for detections # Search for detections
if pi.shape[0]: if pi.shape[0]: