updates
This commit is contained in:
parent
65d70fca78
commit
62d4a74052
|
@ -42,7 +42,7 @@ class LoadImages: # for inference
|
||||||
def __init__(self, path, img_size=416):
|
def __init__(self, path, img_size=416):
|
||||||
files = []
|
files = []
|
||||||
if os.path.isdir(path):
|
if os.path.isdir(path):
|
||||||
files = sorted(glob.glob('%s/*.*' % path))
|
files = sorted(glob.glob(os.path.join(path, '*.*')))
|
||||||
elif os.path.isfile(path):
|
elif os.path.isfile(path):
|
||||||
files = [path]
|
files = [path]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue