Unverified 提交 56f5cb5a authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Print hubconf.py args (#8800)

上级 91112462
...@@ -128,12 +128,13 @@ if __name__ == '__main__': ...@@ -128,12 +128,13 @@ if __name__ == '__main__':
import numpy as np import numpy as np
from PIL import Image from PIL import Image
from utils.general import cv2 from utils.general import cv2, print_args
# Argparser # Argparser
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--model', type=str, default='yolov5s', help='model name') parser.add_argument('--model', type=str, default='yolov5s', help='model name')
opt = parser.parse_args() opt = parser.parse_args()
print_args(vars(opt))
# Model # Model
model = _create(name=opt.model, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True) model = _create(name=opt.model, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论