zishuo/yolov7_plate
1
0
Fork 0
mirror of https://github.com/we0091234/yolov7_plate.git synced 2025-09-26 21:01:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

ncnn_export

Browse Source
This commit is contained in:
we0091234
2023-02-17 19:34:39 +08:00
parent 623fb642c8
commit cf7c50d2e1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ncnn/ncnn_export.py
View File

@@ -27,7 +27,7 @@ from utils.torch_utils import select_device
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--weights', type=str, default='runs/train/yolov714/weights/best.pt', help='weights path')
parser.add_argument('--img-size', nargs='+', type=int, default=[320, 320], help='image size') # height, width
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='image size') # height, width
parser.add_argument('--batch-size', type=int, default=1, help='batch size')
parser.add_argument('--grid', action='store_true', help='export Detect() layer grid')
parser.add_argument('--device', default='cpu', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')