mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Add isdocker() (#2232)

Browse Source 
* Add isdocker()

* Update general.py

* Update general.py
This commit is contained in:
Glenn Jocher 2021-02-16 15:58:07 -08:00 committed by GitHub
parent 5a40ce65ce
commit d2e754b67b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/general.py
View File

@ -47,6 +47,11 @@ def get_latest_run(search_dir='.'):
return max(last_list, key=os.path.getctime) if last_list else '' return max(last_list, key=os.path.getctime) if last_list else ''
def isdocker():
# Is environment a Docker container
return Path('/workspace').exists() # or Path('/.dockerenv').exists()
def check_online(): def check_online():
# Check internet connectivity # Check internet connectivity
import socket import socket
@ -62,7 +67,7 @@ def check_git_status():
print(colorstr('github: '), end='') print(colorstr('github: '), end='')
try: try:
assert Path('.git').exists(), 'skipping check (not a git repository)' assert Path('.git').exists(), 'skipping check (not a git repository)'
assert not Path('/workspace').exists(), 'skipping check (Docker image)' # not Path('/.dockerenv').exists() assert not isdocker(), 'skipping check (Docker image)'
assert check_online(), 'skipping check (offline)' assert check_online(), 'skipping check (offline)'
cmd = 'git fetch && git config --get remote.origin.url' cmd = 'git fetch && git config --get remote.origin.url'
@ -98,13 +103,14 @@ def check_img_size(img_size, s=32):
def check_imshow(): def check_imshow():
# Check if environment supports image displays # Check if environment supports image displays
try: try:
assert not isdocker(), 'cv2.imshow() is disabled in Docker environments'
cv2.imshow('test', np.zeros((1, 1, 3))) cv2.imshow('test', np.zeros((1, 1, 3)))
cv2.waitKey(1) cv2.waitKey(1)
cv2.destroyAllWindows() cv2.destroyAllWindows()
cv2.waitKey(1) cv2.waitKey(1)
return True return True
except Exception as e: except Exception as e:
print(f'WARNING: Environment does not support cv2.imshow() or PIL Image.show() image previews\n{e}') print(f'WARNING: Environment does not support cv2.imshow() or PIL Image.show() image displays\n{e}')
return False return False