mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update hubconf.py attempt_load import (#4428)

Browse Source
This commit is contained in:
Omid Sadeghnezhad 2021-08-16 13:28:02 +04:30 committed by GitHub
parent 19d03a955c
commit dbc06ce292
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
hubconf.py
View File

@ -27,7 +27,8 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
""" """
from pathlib import Path from pathlib import Path
from models.yolo import Model, attempt_load from models.yolo import Model
from models.experimental import attempt_load
from utils.general import check_requirements, set_logging from utils.general import check_requirements, set_logging
from utils.downloads import attempt_download from utils.downloads import attempt_download
from utils.torch_utils import select_device from utils.torch_utils import select_device