def run_RawStruck(seq, rp, bSaveImage):
x = seq.init_rect[0] - 1
y = seq.init_rect[1] - 1
w = seq.init_rect[2]
h = seq.init_rect[3]
path = './results/'
if not os.path.exists(path):
os.makedirs(path)
# x, y, w, h -> initial bounding box
# seq.s_frames is a list of the images
features = "hogANDhist"
kernel = "int"
filter = 0
tracker = RobStruck()
tracker.createTracker(kernel, features, filter)
tracker.initialize(str(seq.s_frames[0]), int(x), int(y), int(w), int(h))
tic = time.clock()
res = np.zeros((len(seq.s_frames),4))
res[0] = [x,y,w,h]
for i in range(1, len(seq.s_frames)):
r = tracker.track(str(seq.s_frames[i]))
res[i] = [r[0], r[1], r[2], r[3]]
duration = time.clock() - tic
result = dict()
result['res'] = res
result['type'] = 'rect'
result['fps'] = round(seq.len / duration, 3)
return result
def run_RawDeepStruck(seq, rp, bSaveImage):
x = seq.init_rect[0] - 1
y = seq.init_rect[1] - 1
w = seq.init_rect[2]
h = seq.init_rect[3]
path = './results/'
if not os.path.exists(path):
os.makedirs(path)
# x, y, w, h -> initial bounding box
# seq.s_frames is a list of the images
features = "deep"
kernel = "linear"
filter = 0
nRadial = 5;
nAngular = 16;
tracker = RobStruck()
dataFolder = '/udrive/student/ibogun2010/Research/Code/DeepAntrack/data/'
#dataFolder = "/Users/Ivan/Code/Tracking/DeepAntrack/data/"
tracker.deepFeatureParams(dataFolder)
tracker.createTracker(kernel, features, filter)
tracker.setLocationSamplerParameters(nRadial, nAngular)
print str(seq.s_frames[0])
tracker.initialize(str(seq.s_frames[0]), int(x), int(y), int(w), int(h))
tracker.setDisplay(0)
tic = time.clock()
res = np.zeros((len(seq.s_frames),4))
res[0] = [x,y,w,h]
for i in range(1, len(seq.s_frames)):
print i, " / ", len(seq.s_frames)
r = tracker.track(str(seq.s_frames[i]))
res[i] = [r[0], r[1], r[2], r[3]]
duration = time.clock() - tic
result = dict()
result['res'] = res
result['type'] = 'rect'
result['fps'] = round(seq.len / duration, 3)
return result
