def trackCamera(conf):
cap = cv2.VideoCapture(0)
if not cap.isOpened():
print 'Error: the video capture do not work.'
return False
ret, frame = cap.read()
if ret is True:
scaleW, scaleH = float(conf.frameWidth) / frame.shape[1], float(
conf.frameHeight) / frame.shape[0]
initBB = Rect(int(conf.frameWidth / 2 - kLiveBoxWidth / 2),
int(conf.frameHeight / 2 - kLiveBoxHeight / 2),
int(kLiveBoxWidth), int(kLiveBoxHeight))
print 'Press "i" to initialize tracker'
else:
print 'Error: video capture invalid.'
return False
if not conf.quietMode:
cv2.namedWindow('Result')
tracker = Tracker(conf)
startFrame, endFrame, paused, doInitialize, count = 0, MAXINT, True, False, 0
result = np.zeros((conf.frameHeight, conf.frameWidth, 3), np.uint8)
random.seed(conf.seed)
for frameid in xrange(startFrame, endFrame + 1):
ret, frameOrig = cap.read()
if ret is True:
frame = cv2.resize(frameOrig, (conf.frameWidth, conf.frameHeight))
frame = cv2.flip(frame, 1)
result = frame[:]
if doInitialize:
if tracker.IsInitialized():
tracker.Reset()
else:
tracker.Initialize(frame, initBB)
doInitialize = False
else:
if not tracker.IsInitialized():
rectangle(result, initBB, WHITE)
else:
print 'Error: video capture invalid.'
return False
if tracker.IsInitialized():
tracker.Track(frame)
# if !conf.quietMode and conf.debugMode:
# tracker.Debug()
box = tracker.GetBox()
rectangle(result, box, GREEN)
# print "hello image"
if not conf.quietMode:
# display menu
if frameid == startFrame:
print("#########################################")
print("MENU")
print("Press \"i\" to start tracking in camera mode")
print("Press \"q\" or esc to escape")
print("Press \"p\" to keep tracking")
print("Press any other key to track 1 frame")
print("#########################################")
# print 'show image'
cv2.imshow("result", result)
if paused:
key = cv2.waitKey() & 0xFF
else:
key = cv2.waitKey(1) & 0xFF
if key != -1:
if key == 27 or key == 81 or key == 113: # esc q
break
elif key == 80 or key == 112: # p
paused = not paused
elif key == 73 or key == 105: # i
doInitialize = True
print 'doInitialize'
if frameid == endFrame:
print 'End of sequence, press any key to exit.'
cv2.waitKey()
cap.release()
cv2.destroyAllWindows()
return True
def trackFrame(conf):
frameFilePath = './' + conf.sequenceBasePath + '/' + conf.sequenceName + '/frame.txt'
if isFile(frameFilePath, 'frame') is False:
return False
fframe = open(frameFilePath, 'r')
line = fframe.readline()
words = filter(None, re.split(r'(?:,|;|\s)\s*', line))
#print words
if len(words) != 2 or int(words[0]) < 0 or int(words[0]) > int(words[1]):
print 'Error: do not get the correct frame params.'
return False
startFrame, endFrame = int(words[0]), int(words[1])
fframe.close()
# Get the info about the image
imgFormatPath = './' + conf.sequenceBasePath + '/' + conf.sequenceName + '/img/0001.jpg'
if isFile(imgFormatPath, 'imgFormat') is False:
return False
img = cv2.imread(imgFormatPath, 0)
scaleW = float(conf.frameWidth) / img.shape[1]
scaleH = float(conf.frameHeight) / img.shape[0]
# Get the groud truth rect
rectFilePath = './' + conf.sequenceBasePath + '/' + conf.sequenceName + '/groundtruth_rect.txt'
if isFile(rectFilePath, 'rect') is False:
return False
rRect = getTrueRect(rectFilePath, startFrame)
initBB = Rect(rRect.getX() * scaleW,
rRect.getY() * scaleH,
rRect.getWidth() * scaleW,
rRect.getHeight() * scaleH)
resultsPath = './' + conf.resultsPath
if isFile(resultsPath, 'res') is False:
return False
fres = open(resultsPath, 'w')
if not conf.quietMode:
cv2.namedWindow('Result')
tracker = Tracker(conf)
paused, doInitialize, count, totalOverlap = True, False, 0, 0.0
random.seed(conf.seed)
result = np.zeros((conf.frameHeight, conf.frameWidth, 3), np.uint8)
rectFilePath = './' + conf.sequenceBasePath + '/' + conf.sequenceName + '/groundtruth_rect.txt'
for frameid in xrange(startFrame, endFrame + 1):
print("Processing frame " + str(frameid) + "/" + str(endFrame))
imgFramePath = './' + conf.sequenceBasePath + '/' + conf.sequenceName + (
'/img/%04d.jpg' % int(frameid))
if isFile(imgFramePath, 'imgFrame') is False:
return False
frameOrig = cv2.imread(imgFramePath, 0)
if frameOrig is None:
print 'Error: do not get valid input frame image.'
return False
frame = cv2.resize(frameOrig, (conf.frameWidth, conf.frameHeight))
result = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)
if frameid == startFrame:
# initBB.printStr()
tracker.Initialize(frame, initBB)
if tracker.IsInitialized():
tracker.Track(frame)
# if !conf.quietMode and conf.debugMode:
# tracker.Debug()
box = tracker.GetBox()
rectangle(result, box, GREEN)
fRect = Rect(box.getX() / scaleW,
box.getY() / scaleH,
box.getWidth() / scaleW,
box.getHeight() / scaleH)
fres.write('%.2f %.2f %.2f %.2f' %
(fRect.getX(), fRect.getY(), fRect.getWidth(),
fRect.getHeight()))
tRect = getTrueRect(rectFilePath, frameid)
overlap = float(tRect.overlap(fRect))
# print 'true Rect: ',
# tRect.printStr()
# print 'tracker Rect: ',
# fRect.printStr()
print 'overlap: %.2f' % overlap
count, totalOverlap = count + 1, totalOverlap + overlap
fres.write(' %.2f\n' % overlap)
# print "hello image"
if not conf.quietMode:
# display menu
if frameid == startFrame:
print("#########################################")
print("MENU")
print("Press \"i\" to start tracking in camera mode")
print("Press \"q\" or esc to escape")
print("Press \"p\" to keep tracking")
print("Press any other key to track 1 frame")
print("#########################################")
# print 'show image'
cv2.imshow("result", result)
if paused:
key = cv2.waitKey() & 0xFF
else:
key = cv2.waitKey(1) & 0xFF
if key != -1:
if key == 27 or key == 81 or key == 113: # esc q
break
elif key == 80 or key == 112: # p
paused = not paused
elif key == 73 or key == 105: # i
doInitialize = True
print 'doInitialize'
if frameid == endFrame:
print 'End of sequence, press any key to exit.'
cv2.waitKey()
score = totalOverlap / float(count)
print("")
print("######################################")
print("Final Overlap Score is: " + str(score))
print("######################################")
print("")
fres.write('%.3f\n' % score)
if not fres.closed:
fres.close()
cv2.destroyAllWindows()
return True
