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

updates

This commit is contained in:
Glenn Jocher 2019-10-01 16:04:56 +02:00
parent 9a48f23726
commit 84f0df6c34
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
utils/utils.py
View File

@ -224,6 +224,7 @@ def compute_ap(recall, precision):
# Returns # Returns
The average precision as computed in py-faster-rcnn. The average precision as computed in py-faster-rcnn.
""" """
# Append sentinel values to beginning and end # Append sentinel values to beginning and end
mrec = np.concatenate(([0.], recall, [1.])) mrec = np.concatenate(([0.], recall, [1.]))
mpre = np.concatenate(([0.], precision, [0.])) mpre = np.concatenate(([0.], precision, [0.]))
@ -232,11 +233,15 @@ def compute_ap(recall, precision):
for i in range(mpre.size - 1, 0, -1): for i in range(mpre.size - 1, 0, -1):
mpre[i - 1] = np.maximum(mpre[i - 1], mpre[i]) mpre[i - 1] = np.maximum(mpre[i - 1], mpre[i])
# Calculate area under PR curve, looking for points where x axis (recall) changes # Integrate area under curve
i = np.where(mrec[1:] != mrec[:-1])[0] method = 'interp' # methods: 'continuous', 'interp'
if method == 'interp':
x = np.linspace(0, 1, 101) # 101-point interp (COCO)
ap = np.trapz(np.interp(x, mrec, mpre), x) # integrate
else: # 'continuous'
i = np.where(mrec[1:] != mrec[:-1])[0] # points where x axis (recall) changes
ap = np.sum((mrec[i + 1] - mrec[i]) * mpre[i + 1]) # area under curve
# Sum (\Delta recall) * prec
ap = np.sum((mrec[i + 1] - mrec[i]) * mpre[i + 1])
return ap return ap