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