def setup_config(config_path):
cfg = get_config()
cfg.merge_from_file(config_path)
return cfg
default="./configs/yolov3.yaml")
parser.add_argument("--config_deepsort",
type=str,
default="./configs/deep_sort.yaml")
# parser.add_argument("--ignore_display", dest="display", action="store_false", default=True)
parser.add_argument("--display", action="store_true")
parser.add_argument("--frame_interval", type=int, default=1)
parser.add_argument("--display_width", type=int, default=800)
parser.add_argument("--display_height", type=int, default=600)
parser.add_argument("--save_path", type=str, default="./output/")
parser.add_argument("--cpu",
dest="use_cuda",
action="store_false",
default=True)
parser.add_argument("--camera",
action="store",
dest="cam",
type=int,
default="-1")
return parser.parse_args()
if __name__ == "__main__":
args = parse_args()
cfg = get_config()
cfg.merge_from_file(args.config_detection)
cfg.merge_from_file(args.config_deepsort)
with VideoTracker(cfg, args, video_path=args.VIDEO_PATH) as vdo_trk:
vdo_trk.run()
def get_merged_config():
cfg = get_config()
cfg.merge_from_file(join(dirname(__file__), 'configs', 'deep_sort.yaml'))
return cfg