7eb23e3c1d
* Initial commit * Initial commit * Cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix precommit errors * Remove TF builds from CI * export last.pt * Created using Colaboratory * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
19 lines
445 B
Python
19 lines
445 B
Python
# 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
|