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

updates

This commit is contained in:
Glenn Jocher 2019-12-24 12:42:22 -08:00
parent 804f82a4b0
commit 0f225afe33
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test.py
View File

@ -106,6 +106,9 @@ def test(cfg,
# with open('test.txt', 'a') as file: # with open('test.txt', 'a') as file:
# [file.write('%11.5g' * 7 % tuple(x) + '\n') for x in pred] # [file.write('%11.5g' * 7 % tuple(x) + '\n') for x in pred]
# Clip boxes to image bounds
clip_coords(pred, (height, width))
# Append to pycocotools JSON dictionary # Append to pycocotools JSON dictionary
if save_json: if save_json:
# [{"image_id": 42, "category_id": 18, "bbox": [258.15, 41.29, 348.26, 243.78], "score": 0.236}, ... # [{"image_id": 42, "category_id": 18, "bbox": [258.15, 41.29, 348.26, 243.78], "score": 0.236}, ...
@ -120,11 +123,8 @@ def test(cfg,
'bbox': [floatn(x, 3) for x in box[di]], 'bbox': [floatn(x, 3) for x in box[di]],
'score': floatn(d[4], 5)}) 'score': floatn(d[4], 5)})
# Clip boxes to image bounds
clip_coords(pred, (height, width))
# Assign all predictions as incorrect # Assign all predictions as incorrect
correct = torch.zeros(len(pred), niou) correct = torch.zeros(len(pred), niou, dtype=torch.bool)
if nl: if nl:
detected = [] # target indices detected = [] # target indices
tcls_tensor = labels[:, 0] tcls_tensor = labels[:, 0]
@ -147,7 +147,7 @@ def test(cfg,
d = ti[i[j]] # detected target d = ti[i[j]] # detected target
if d not in detected: if d not in detected:
detected.append(d) detected.append(d)
correct[pi[j]] = (ious[j] > iou_thres).float() # iou_thres is 1xn correct[pi[j]] = ious[j] > iou_thres # iou_thres is 1xn
if len(detected) == nl: # all targets already located in image if len(detected) == nl: # all targets already located in image
break break