diff --git a/utils/datasets.py b/utils/datasets.py index e88ce76..cc5411f 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -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)