glenn-jocher
commited on
Commit
•
1e7674d
1
Parent(s):
a21bd06
Update export.py --img-size comment
Browse files- 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
|