thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #169 from youngjae-avikus/master

Browse Source 
fix error in getting args(width,height,channel) as type str
This commit is contained in:
Marcos Luciano
2022-05-10 12:25:37 -03:00
committed by GitHub
parent 4cae635529 6246d4b3c6
commit 4192a16751
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
utils/gen_wts_yoloV5.py
View File

@@ -65,9 +65,9 @@ def parse_args():
parser = argparse.ArgumentParser(description="PyTorch YOLOv5 conversion")
parser.add_argument("-w", "--weights", required=True, help="Input weights (.pt) file path (required)")
parser.add_argument("-c", "--yaml", help="Input cfg (.yaml) file path")
parser.add_argument("-mw", "--width", help="Model width (default = 640 / 1280 [P6])")
parser.add_argument("-mh", "--height", help="Model height (default = 640 / 1280 [P6])")
parser.add_argument("-mc", "--channels", help="Model channels (default = 3)")
parser.add_argument("-mw", "--width", type=int, help="Model width (default = 640 / 1280 [P6])")
parser.add_argument("-mh", "--height", type=int, help="Model height (default = 640 / 1280 [P6])")
parser.add_argument("-mc", "--channels", type=int, help="Model channels (default = 3)")
parser.add_argument("--p6", action="store_true", help="P6 model")
args = parser.parse_args()
if not os.path.isfile(args.weights):