xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix *.txt source inference dataloader (#4468)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
GrunCrow 2023-08-21 13:59:19 +02:00 committed by GitHub
parent 3acead7e79
commit c659c0fa7b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_python.py
View File

@ -49,6 +49,16 @@ def test_predict_dir():
model(source=ASSETS, imgsz=32) model(source=ASSETS, imgsz=32)
def test_predict_txt():
# Write a list of sources to a txt file
txt_file = TMP / 'sources.txt'
with open(txt_file, 'w') as f:
for x in [ASSETS / 'bus.jpg', ASSETS / 'zidane.jpg']:
f.write(f'{x}\n')
model = YOLO(MODEL)
model(source=txt_file, imgsz=640)
def test_predict_img(): def test_predict_img():
model = YOLO(MODEL) model = YOLO(MODEL)
seg_model = YOLO(WEIGHTS_DIR / 'yolov8n-seg.pt') seg_model = YOLO(WEIGHTS_DIR / 'yolov8n-seg.pt')

2
ultralytics/data/loaders.py
View File

@ -182,7 +182,7 @@ class LoadImages:
parent = None parent = None
if isinstance(path, str) and Path(path).suffix == '.txt': # *.txt file with img/vid/dir on each line if isinstance(path, str) and Path(path).suffix == '.txt': # *.txt file with img/vid/dir on each line
parent = Path(path).parent parent = Path(path).parent
path = Path(path).read_text().rsplit() path = Path(path).read_text().splitlines() # list of sources
files = [] files = []
for p in sorted(path) if isinstance(path, (list, tuple)) else [path]: for p in sorted(path) if isinstance(path, (list, tuple)) else [path]:
a = str(Path(p).absolute()) # do not use .resolve() https://github.com/ultralytics/ultralytics/issues/2912 a = str(Path(p).absolute()) # do not use .resolve() https://github.com/ultralytics/ultralytics/issues/2912