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
|