mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
7ebb5e5da6
* Update __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * notebook_init * notebook_init * notebook_init * notebook_init * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * notebook_init * Created using Colaboratory * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
19 lines
476 B
Python
19 lines
476 B
Python
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
|
"""
|
|
utils/initialization
|
|
"""
|
|
|
|
|
|
def notebook_init():
|
|
# For YOLOv5 notebooks
|
|
print('Checking setup...')
|
|
from IPython import display # to display images and clear console output
|
|
|
|
from utils.general import emojis
|
|
from utils.torch_utils import select_device # YOLOv5 imports
|
|
|
|
display.clear_output()
|
|
select_device(newline=False)
|
|
print(emojis('Setup complete ✅'))
|
|
return display
|