@@ -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) | |||