[f26a44]: / yolov5 / utils / __init__.py

Download this file

19 lines (14 with data), 458 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
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