Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enhanced check_requirements() with auto-install #2575

Merged
merged 5 commits into from
Mar 24, 2021
Merged

Enhanced check_requirements() with auto-install #2575

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
6207da2
Update check_requirements() with auto-install
glenn-jocher Mar 23, 2021
ad272cc
cleanup
glenn-jocher Mar 23, 2021
4a61912
cleanup 2
glenn-jocher Mar 23, 2021
09982f3
Check requirements.txt file exists
glenn-jocher Mar 23, 2021
6e777ec
cleanup 3
glenn-jocher Mar 23, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 15 additions & 5 deletions utils/general.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# General utils
# YOLOv5 general utils

import glob
import logging
Expand Down Expand Up @@ -86,10 +86,20 @@ def check_git_status():

def check_requirements(file='requirements.txt', exclude=()):
# Check installed dependencies meet requirements
import pkg_resources
requirements = [f'{x.name}{x.specifier}' for x in pkg_resources.parse_requirements(Path(file).open())
if x.name not in exclude]
pkg_resources.require(requirements) # DistributionNotFound or VersionConflict exception if requirements not met
import pkg_resources as pkg
prefix = colorstr('red', 'bold', 'requirements:')
file = Path(file)
if not file.exists():
print(f"{prefix} {file.resolve()} not found, check failed.")
return

requirements = [f'{x.name}{x.specifier}' for x in pkg.parse_requirements(file.open()) if x.name not in exclude]
for r in requirements:
try:
pkg.require(r)
except Exception as e: # DistributionNotFound or VersionConflict if requirements not met
print(f"{prefix} {e.req} not found and is required by YOLOv5, attempting auto-install...")
print(subprocess.check_output(f"pip install '{e.req}'", shell=True).decode())


def check_img_size(img_size, s=32):
Expand Down