From dee692baf0bcf9c7d6ce5a435c31e78939c8f8d0 Mon Sep 17 00:00:00 2001 From: Zhongdao Date: Wed, 29 Jan 2020 23:56:24 +0800 Subject: [PATCH] Modify demo.py --- demo.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/demo.py b/demo.py index 1d4a83b..32b9917 100644 --- a/demo.py +++ b/demo.py @@ -12,6 +12,7 @@ from utils.io import read_results from utils.log import logger from utils.timer import Timer from utils.evaluation import Evaluator +from utils.parse_config import parse_model_cfg import utils.datasets as datasets import torch from track import eval_seq @@ -22,6 +23,9 @@ def track(opt): result_root = opt.output_root if opt.output_root!='' else '.' mkdir_if_missing(result_root) + cfg_dict = parse_model_cfg(opt.cfg) + opt.img_size = [int(cfg_dict[0]['width']), int(cfg_dict[0]['height'])] + # run tracking timer = Timer() accs = [] @@ -48,7 +52,6 @@ if __name__ == '__main__': parser = argparse.ArgumentParser(prog='demo.py') parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') parser.add_argument('--weights', type=str, default='weights/latest.pt', help='path to weights file') - parser.add_argument('--img-size', type=int, default=(1088, 608), help='size of each image dimension') parser.add_argument('--iou-thres', type=float, default=0.5, help='iou threshold required to qualify as detected') parser.add_argument('--conf-thres', type=float, default=0.5, help='object confidence threshold') parser.add_argument('--nms-thres', type=float, default=0.4, help='iou threshold for non-maximum suppression')