소스 검색

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

5.0
Glenn Jocher 4 년 전
부모
커밋
8412d44dce
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. +5
    -3
      utils/datasets.py

+ 5
- 3
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)


Loading…
취소
저장