From 53294f28b4b23cec757b8fb9c704f7ae436b854c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 23 Jun 2020 10:53:42 -0700 Subject: [PATCH] update yolo.py --- models/yolo.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/models/yolo.py b/models/yolo.py index e491e64..158a241 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -1,7 +1,5 @@ import argparse -import yaml - from models.experimental import * @@ -206,7 +204,7 @@ def parse_model(md, ch): # model_dict, input_channels(3) if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('--cfg', type=str, default='yolov5s.yaml', help='model.yaml') + parser.add_argument('--cfg', type=str, default='yolov5m-p6.yaml', help='model.yaml') parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') opt = parser.parse_args() opt.cfg = check_file(opt.cfg) # check file