2021-11-14 22:22:59 +01:00
|
|
|
# YOLOv3 🚀 by Ultralytics, GPL-3.0 license
|
|
|
|
|
"""
|
|
|
|
|
utils/initialization
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def notebook_init():
|
|
|
|
|
# For 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 # imports
|
|
|
|
|
|
|
|
|
|
display.clear_output()
|
|
|
|
|
select_device(newline=False)
|
|
|
|
|
print(emojis('Setup complete ✅'))
|
|
|
|
|
return display
|
