2583f842b8
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
22 lines
626 B
Python
22 lines
626 B
Python
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
|
|
|
from ultralytics.utils import ASSETS, ROOT, WEIGHTS_DIR, checks, is_dir_writeable
|
|
|
|
# Constants used in tests
|
|
MODEL = WEIGHTS_DIR / "path with spaces" / "yolov8n.pt" # test spaces in path
|
|
CFG = "yolov8n.yaml"
|
|
SOURCE = ASSETS / "bus.jpg"
|
|
TMP = (ROOT / "../tests/tmp").resolve() # temp directory for test files
|
|
IS_TMP_WRITEABLE = is_dir_writeable(TMP)
|
|
CUDA_IS_AVAILABLE = checks.cuda_is_available()
|
|
CUDA_DEVICE_COUNT = checks.cuda_device_count()
|
|
|
|
__all__ = (
|
|
"MODEL",
|
|
"CFG",
|
|
"SOURCE",
|
|
"TMP",
|
|
"IS_TMP_WRITEABLE",
|
|
"CUDA_IS_AVAILABLE",
|
|
"CUDA_DEVICE_COUNT",
|
|
)
|