|
|
|
@ -20,8 +20,8 @@ import torch
|
|
|
|
|
from matplotlib import font_manager
|
|
|
|
|
|
|
|
|
|
from ultralytics.yolo.utils import (AUTOINSTALL, LOGGER, ONLINE, ROOT, USER_CONFIG_DIR, TryExcept, clean_url, colorstr,
|
|
|
|
|
downloads, emojis, is_colab, is_docker, is_kaggle, is_online, is_pip_package,
|
|
|
|
|
url2file)
|
|
|
|
|
downloads, emojis, is_colab, is_docker, is_jupyter, is_kaggle, is_online,
|
|
|
|
|
is_pip_package, url2file)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def is_ascii(s) -> bool:
|
|
|
|
@ -325,8 +325,11 @@ def check_yolo(verbose=True, device=''):
|
|
|
|
|
"""Return a human-readable YOLO software and hardware summary."""
|
|
|
|
|
from ultralytics.yolo.utils.torch_utils import select_device
|
|
|
|
|
|
|
|
|
|
if is_colab():
|
|
|
|
|
shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory
|
|
|
|
|
if is_jupyter():
|
|
|
|
|
if check_requirements('wandb', install=False):
|
|
|
|
|
os.system('pip uninstall -y wandb') # uninstall wandb: unwanted account creation prompt with infinite hang
|
|
|
|
|
if is_colab():
|
|
|
|
|
shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory
|
|
|
|
|
|
|
|
|
|
if verbose:
|
|
|
|
|
# System info
|
|
|
|
|