Add notebook_init() to utils/__init__.py (#5488)
* 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>
This commit is contained in:
1 parent
7207fe95e5
commit
7ebb5e5da6
3 files changed
+35
-17
No files matched your search
@@ -0,0 +1,18 @@
|
||||
# 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
|
||||
Reference in new issue
Block a user