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

update datasets.py to add detect.py --source glob_command feature

Browse Source
This commit is contained in:
Glenn Jocher 2020-07-12 14:22:15 -07:00
parent eae33303d6
commit 8412d44dce
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/datasets.py
View File

@ -70,10 +70,12 @@ class LoadImages: # for inference
def __init__(self, path, img_size=640):
p = str(Path(path)) # os-agnostic
p = os.path.abspath(p) # absolute path
if os.path.isdir(p):
files = sorted(glob.glob(os.path.join(p, '*.*')))
if '*' in p:
files = sorted(glob.glob(p)) # glob
elif os.path.isdir(p):
files = sorted(glob.glob(os.path.join(p, '*.*'))) # dir
elif os.path.isfile(p):
files = [p]
files = [p] # files
else:
raise Exception('ERROR: %s does not exist' % p)