2021-11-14 22:22:59 +01:00
|
|
|
# YOLOv3 🚀 by Ultralytics, GPL-3.0 license
|
|
|
|
|
"""
|
|
|
|
|
utils/initialization
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
2023-02-21 21:43:51 +05:30
|
|
|
def notebook_init():
|
|
|
|
|
# For notebooks
|
2021-11-14 22:22:59 +01:00
|
|
|
print('Checking setup...')
|
2023-02-11 15:20:10 +04:00
|
|
|
from IPython import display # to display images and clear console output
|
|
|
|
|
|
2023-02-21 21:43:51 +05:30
|
|
|
from utils.general import emojis
|
|
|
|
|
from utils.torch_utils import select_device # imports
|
2023-02-11 15:20:10 +04:00
|
|
|
|
2023-02-21 21:43:51 +05:30
|
|
|
display.clear_output()
|
2021-11-14 22:22:59 +01:00
|
|
|
select_device(newline=False)
|
2023-02-21 21:43:51 +05:30
|
|
|
print(emojis('Setup complete ✅'))
|
2021-11-14 22:22:59 +01:00
|
|
|
return display
|