yolov5/utils/__init__.py

# 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