Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

fix channel dim selection on segmentation target #1509

Merged
Prev Previous commit
Next Next commit
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
pre-commit-ci[bot] committed Jan 9, 2023
commit cf65734aa7cecd696104d9e4223f4f5075f2c205
8 changes: 2 additions & 6 deletions tests/image/segmentation/test_data.py
Original file line number Diff line number Diff line change
@@ -92,13 +92,9 @@ def per_batch_transform(self) -> Callable:
os.makedirs(str(tmp_dir / "images"))
os.makedirs(str(tmp_dir / "targets"))

images = [
str(tmp_dir / "images" / "img1.png")
]
images = [str(tmp_dir / "images" / "img1.png")]

targets = [
str(tmp_dir / "targets" / "img1.png")
]
targets = [str(tmp_dir / "targets" / "img1.png")]

num_classes: int = 2
img_size: Tuple[int, int] = (128, 128)