def get_results(net_type, net_base_path, model_name, vis=False): # get the default parameters p = Config() # load all videos all_videos = os.listdir(p.seq_base_path) p.bbox_output = True p.bbox_output_path = os.path.join("./tracking_result/", str(net_type)) p.visualization = vis if p.bbox_output: if not os.path.exists(p.bbox_output_path): os.makedirs(p.bbox_output_path) fps_all = .0 for it, video in enumerate(all_videos): p.video = video print("Processing %s ... " % p.video) print("{} / {}".format(it, len(all_videos))) bbox_result, fps = run_tracker(p, net_type, net_base_path, model_name) # fps for this video print("FPS: %d " % fps) # saving tracking results if p.bbox_output: np.savetxt(os.path.join(p.bbox_output_path, p.video.lower() + '_SiamFC.txt'), bbox_result, fmt='%.3f') fps_all = fps_all + fps avg_fps = fps_all / len(all_videos) print("Average FPS: %f" % avg_fps)
def get_specific_video_results(net_type, net_base_path, model_name, video_name, vis=False, save_to_file=False): # get the default parameters p = Config() p.visualization = vis p.save_to_file = save_to_file p.save_to_video = False p.video = video_name print("Processing %s ... " % p.video) bbox_result, fps = run_tracker(p, net_type, net_base_path, model_name) # fps for this video print("FPS: %d " % fps)