Elgato_dark/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics synced 2026-04-21 14:07:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Remove cached *.npy files in grayscale tests (#23554)

Browse source
This commit is contained in:
Jing Qiu 2026-02-04 22:37:08 +08:00 committed by GitHub
parent aacb357fbe
commit 1da4e93318
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_python.py
View file

@ -796,6 +796,10 @@ def test_grayscale(task: str, model: str, data: str, tmp_path) -> None:
model = YOLO(model)
model.train(data=grayscale_data, epochs=1, imgsz=32, close_mosaic=1, cache="disk")
# remove npy files in train/val splits if exists, avoiding interference with other tests
for split in {"train", "val"}:
for npy_file in (Path(data["path"]) / data[split]).glob("*.npy"):
npy_file.unlink()
model.val(data=grayscale_data)
im = np.zeros((32, 32, 1), dtype=np.uint8)
model.predict(source=im, imgsz=32, save_txt=True, save_crop=True, augment=True)