kepler62f
GitHub
3 years ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
test.py
View File
| # 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 | |||||
| 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 | # Search for detections | ||||
| if pi.shape[0]: | if pi.shape[0]: |
Loading…