Simplify check_requirements() usage (#4855)

* Simplify `check_requirements()` usage

* remove assert, print()
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-18 18:34:30 +02:00
1 parent 4d1a2ac87e
commit 4c839eeb10
6 files changed
+9 -6

No files matched your search

+4 -1
View File
@@ -37,6 +37,9 @@ pd.options.display.max_columns = 10
cv2.setNumThreads(0) # prevent OpenCV from multithreading (incompatible with PyTorch DataLoader)
os.environ['NUMEXPR_MAX_THREADS'] = str(min(os.cpu_count(), 8)) # NumExpr max threads
FILE = Path(__file__).resolve()
ROOT = FILE.parents[1] # YOLOv5 root directory
class Profile(contextlib.ContextDecorator):
# Usage: @Profile() decorator or 'with Profile():' context manager
@@ -222,7 +225,7 @@ def check_version(current='0.0.0', minimum='0.0.0', name='version ', pinned=Fals
@try_except
def check_requirements(requirements='requirements.txt', exclude=(), install=True):
def check_requirements(requirements=ROOT / 'requirements.txt', exclude=(), install=True):
# Check installed dependencies meet requirements (pass *.txt file or list of packages)
prefix = colorstr('red', 'bold', 'requirements:')
check_python() # check python version