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 is_colab() function (#3018)

Browse Source
This commit is contained in:
Glenn Jocher 2021-05-03 12:37:50 +02:00 committed by GitHub
parent e2a80c6c0f
commit fcd57028d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
utils/general.py
View File

@ -51,11 +51,20 @@ 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(): def is_docker():
# Is environment a Docker container # Is environment a Docker container
return Path('/workspace').exists() # or Path('/.dockerenv').exists() return Path('/workspace').exists() # or Path('/.dockerenv').exists()
def is_colab():
# Is environment a Google Colab instance
try:
import google.colab
return True
except Exception as e:
return False
def emojis(str=''): def emojis(str=''):
# Return platform-dependent emoji-safe version of string # Return platform-dependent emoji-safe version of string
return str.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else str return str.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else str
@ -81,7 +90,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 isdocker(), 'skipping check (Docker image)' assert not is_docker(), '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'
@ -138,7 +147,8 @@ 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' assert not is_docker(), 'cv2.imshow() is disabled in Docker environments'
assert not is_colab(), 'cv2.imshow() is disabled in Google Colab 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()