Print hubconf.py args (#8800)
This commit is contained in:
1 parent
9111246208
commit
56f5cb5a28
1 file changed
+2
-1
+2
-1
@@ -128,12 +128,13 @@ if __name__ == '__main__':
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
|
||||
from utils.general import cv2
|
||||
from utils.general import cv2, print_args
|
||||
|
||||
# Argparser
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--model', type=str, default='yolov5s', help='model name')
|
||||
opt = parser.parse_args()
|
||||
print_args(vars(opt))
|
||||
|
||||
# Model
|
||||
model = _create(name=opt.model, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True)
|
||||
|
||||
Reference in new issue
Block a user