return result, result_bb, fps, average_overlap
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('-s', '--seq', default='', help='input seq')
parser.add_argument('-j', '--json', default='', help='input json')
parser.add_argument('-f', '--savefig', action='store_true')
parser.add_argument('-d', '--display', action='store_true')
args = parser.parse_args()
assert (args.seq != '' or args.json != '')
# Generate sequence config
img_list, init_bbox, gt, savefig_dir, display, result_path = gen_config(
args)
# Run tracker
result, result_bb, fps, average_overlap = run_mdnet(
img_list, init_bbox, gt=gt, savefig_dir=savefig_dir, display=display)
# Save result
res = {}
res['average'] = average_overlap
res['res'] = result_bb.round().tolist()
res['type'] = 'rect'
res['fps'] = fps
json.dump(res, open(result_path, 'w'), indent=2)
print ("Frame %d/%d, Score %.3f, Time %.3f" % \
(i, len(img_list), target_score, spf))
else:
print ("Frame %d/%d, Overlap %.3f, Score %.3f, Time %.3f" % \
(i, len(img_list), overlap_ratio(gt[i],result[i])[0], target_score, spf))
print(len(img_list), "costs {}".format(time.time()-s))
fps = len(img_list) / spf_total
return result, result_top5, fps
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('-s', '--seq', default='', help='input seq')
parser.add_argument('-l', '--label_id', default=1)
args = parser.parse_args()
# Generate sequence config
img_list, init_bbox, savefig_dir, result_path = gen_config(args.seq, args.label_id)
# Run tracker
result, result_top5, fps = run_mdnet(img_list, init_bbox, gt=None, savefig_dir=savefig_dir)
# Save result
res = {}
res['res'] = result.round().tolist()
res['type'] = 'rect'
res['fps'] = fps
json.dump(res, open(os.path.join(result_path, "result.json"), 'w'), indent=2)