glenn-jocher commited on
Commit
1e7674d
1 Parent(s): a21bd06

Update export.py --img-size comment

Browse files
Files changed (1) hide show
  1. models/export.py +1 -1
models/export.py CHANGED
@@ -17,7 +17,7 @@ from utils.general import set_logging
17
  if __name__ == '__main__':
18
  parser = argparse.ArgumentParser()
19
  parser.add_argument('--weights', type=str, default='./yolov5s.pt', help='weights path') # from yolov5/models/
20
- parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='image size')
21
  parser.add_argument('--batch-size', type=int, default=1, help='batch size')
22
  opt = parser.parse_args()
23
  opt.img_size *= 2 if len(opt.img_size) == 1 else 1 # expand
 
17
  if __name__ == '__main__':
18
  parser = argparse.ArgumentParser()
19
  parser.add_argument('--weights', type=str, default='./yolov5s.pt', help='weights path') # from yolov5/models/
20
+ parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='image size') # height, width
21
  parser.add_argument('--batch-size', type=int, default=1, help='batch size')
22
  opt = parser.parse_args()
23
  opt.img_size *= 2 if len(opt.img_size) == 1 else 1 # expand