zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

YOLOv5 PyTorch Hub models >> check_requirements() (#2591) 

Prints 'Please restart runtime or rerun command for update to take effect.' following package auto-install to inform users to restart/rerun.
This commit is contained in:
Glenn Jocher 2021-03-24 16:42:54 +01:00 committed by GitHub
parent 8ace1b1b99
commit 75feeb797c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils/general.py
View File

@ -100,6 +100,7 @@ def check_requirements(file='requirements.txt', exclude=()):
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())
print(f'Please restart runtime or rerun command for update to take effect.')
def check_img_size(img_size, s=32):