Browse Source

Comment fix (#3058)

Co-authored-by: kepler62f <>
modifyDataloader
kepler62f GitHub 3 years ago
parent
commit
f2de1ad2aa
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      test.py

+ 2
- 2
test.py View File

@@ -188,8 +188,8 @@ def test(data,

# Per target class
for cls in torch.unique(tcls_tensor):
ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1) # prediction indices
pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1) # target indices
ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1) # target indices
pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1) # prediction indices

# Search for detections
if pi.shape[0]:

Loading…
Cancel
Save