Skip to content

Commit

Permalink
Add is_colab() function (ultralytics#3018)
Browse files Browse the repository at this point in the history
(cherry picked from commit fcd5702)
  • Loading branch information
glenn-jocher authored and Lechtr committed May 24, 2021
1 parent daff9ca commit dbd4d7f
Showing 1 changed file with 13 additions and 3 deletions.
16 changes: 13 additions & 3 deletions utils/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,11 +51,20 @@ def get_latest_run(search_dir='.'):
return max(last_list, key=os.path.getctime) if last_list else ''


def isdocker():
def is_docker():
# Is environment a Docker container
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=''):
# Return platform-dependent emoji-safe version of string
return str.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else str
Expand Down Expand Up @@ -84,7 +93,7 @@ def check_git_status():
print(colorstr('github: '), end='')
try:
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)'

cmd = 'git fetch && git config --get remote.origin.url'
Expand Down Expand Up @@ -141,7 +150,8 @@ def check_img_size(img_size, s=32):
def check_imshow():
# Check if environment supports image displays
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.waitKey(1)
cv2.destroyAllWindows()
Expand Down

0 comments on commit dbd4d7f

Please sign in to comment.