generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

allow multiple detectors

Browse Source
This commit is contained in:
marina.kolpakova 2013-01-21 15:53:25 +04:00
parent 469eeea370
commit 11f3927c58
2 changed files with 48 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/sft/misk/roc_test.py
View File

@ -7,6 +7,8 @@ import sys, os, os.path, glob, math, cv2
from datetime import datetime from datetime import datetime
import numpy import numpy
plot_colors = ['b', 'r', 'g', 'c', 'm']
# "key" : ( b, g, r) # "key" : ( b, g, r)
bgr = { "red" : ( 0, 0, 255), bgr = { "red" : ( 0, 0, 255),
"green" : ( 0, 255, 0), "green" : ( 0, 255, 0),
@ -19,7 +21,7 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description = 'Plot ROC curve using Caltech mathod of per image detection performance estimation.') parser = argparse.ArgumentParser(description = 'Plot ROC curve using Caltech mathod of per image detection performance estimation.')
# positional # positional
parser.add_argument("cascade", help = "Path to the tested detector.") parser.add_argument("cascade", help = "Path to the tested detector.", nargs='+')
parser.add_argument("input", help = "Image sequence pattern.") parser.add_argument("input", help = "Image sequence pattern.")
parser.add_argument("annotations", help = "Path to the annotations.") parser.add_argument("annotations", help = "Path to the annotations.")
@ -34,9 +36,13 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
# parse annotations print args.cascade
# # parse annotations
sft.initPlot()
samples = call_parser(args.anttn_format, args.annotations) samples = call_parser(args.anttn_format, args.annotations)
cascade = sft.cascade(args.min_scale, args.max_scale, args.nscales, args.cascade) for idx, each in enumerate(args.cascade):
print each
cascade = sft.cascade(args.min_scale, args.max_scale, args.nscales, each)
pattern = args.input pattern = args.input
camera = cv2.VideoCapture(pattern) camera = cv2.VideoCapture(pattern)
@ -77,4 +83,6 @@ if __name__ == "__main__":
print nframes, nannotated print nframes, nannotated
fppi, miss_rate = sft.computeROC(confidenses, tp, nannotated, nframes) fppi, miss_rate = sft.computeROC(confidenses, tp, nannotated, nframes)
sft.plotLogLog(fppi, miss_rate) sft.plotLogLog(fppi, miss_rate, plot_colors[idx])
sft.showPlot("roc_curve.png")

19
apps/sft/misk/sft.py
View File

@ -55,7 +55,7 @@ def crop_rect(rect, factor):
# #
def plotLogLog(fppi, miss_rate): def initPlot():
fig, ax = plt.subplots() fig, ax = plt.subplots()
fig.canvas.draw() fig.canvas.draw()
@ -63,22 +63,19 @@ def plotLogLog(fppi, miss_rate):
plt.xlabel("fppi") plt.xlabel("fppi")
plt.ylabel("miss rate") plt.ylabel("miss rate")
plt.title("ROC curve Bahnhof") plt.title("ROC curve Bahnhof")
# plt.yticks( [0.05, 0.10, 0.20, 0.30, 0.40, 0.50, 0.64, 0.80])
# ylabels = [item.get_text() for item in ax.get_yticklabels()]
# ax.set_yticklabels( ylabels )
plt.grid(True) plt.grid(True)
# plt.xticks( [pow(10, -4), pow(10, -3), pow(10, -2), pow(10, -1), pow(10, 0), pow(10, 1)])
# xlabels = [item.get_text() for item in ax.get_xticklabels()]
# ax.set_xticklabels( xlabels )
plt.xscale('log') plt.xscale('log')
plt.yscale('log') plt.yscale('log')
plt.semilogy(fppi, miss_rate, color='m', linewidth=2) def showPlot(name):
plt.savefig(name)
plt.show() plt.show()
def plotLogLog(fppi, miss_rate, c):
plt.semilogy(fppi, miss_rate, color = c, linewidth = 2)
def draw_rects(img, rects, color, l = lambda x, y : x + y): def draw_rects(img, rects, color, l = lambda x, y : x + y):
if rects is not None: if rects is not None:
for x1, y1, x2, y2 in rects: for x1, y1, x2, y2 in rects: