Print hubconf.py args ()

pull/8798/head^2
Glenn Jocher 2022-07-30 21:02:26 +02:00 committed by GitHub
parent 9111246208
commit 56f5cb5a28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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)