Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Print hubconf.py args (#8800)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-07-30 21:02:26 +02:00
1 parent 9111246208
commit 56f5cb5a28
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
hubconf.py
+2 -1
View File
@@ -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)