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

updates

This commit is contained in:
Glenn Jocher 2018-09-10 16:26:40 +02:00
parent ba1b3d8fe5
commit 751e02de3e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -80,7 +80,9 @@ for batch_i, (imgs, targets) in enumerate(dataloader):
# If no annotations add number of detections as incorrect # If no annotations add number of detections as incorrect
if annotations.size(0) == 0: if annotations.size(0) == 0:
target_cls = [] target_cls = []
correct.extend([0 for _ in range(len(detections))]) #correct.extend([0 for _ in range(len(detections))])
mAPs.append(0)
continue
else: else:
target_cls = annotations[:, 0] target_cls = annotations[:, 0]