apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

edit in comments (#1417)

Browse Source 
Co-authored-by: Priteshkumar Bharatbhai Gohil <pgohil@assystemtechnologies.com>
This commit is contained in:
priteshgohil 2020-07-27 19:29:45 +02:00 committed by GitHub
parent 8de13f114d
commit e0a5a6b411
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -145,8 +145,8 @@ def test(cfg,
# Per target class
for cls in torch.unique(tcls_tensor):
ti = (cls == tcls_tensor).nonzero().view(-1) # prediction indices
pi = (cls == pred[:, 5]).nonzero().view(-1) # target indices
ti = (cls == tcls_tensor).nonzero().view(-1) # target indices
pi = (cls == pred[:, 5]).nonzero().view(-1) # prediction indices
# Search for detections
if pi.shape[0]: