Modernize data loader, remove read image helper

This commit is contained in:
henryruhs
2025-02-21 01:13:37 +01:00
parent 39c0313202
commit 254bc17c98
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -3,8 +3,8 @@ import random
import cv2
import torch
import transforms from torchvision
from torch.utils.data import Dataset
from torchvision import transforms
from .types import Batch, Paths
+1 -1
View File
@@ -5,8 +5,8 @@ from typing import Tuple
import cv2
import torch
import transforms from torchvision
from torch.utils.data import TensorDataset
from torchvision import transforms
from .types import Batch, ImagePathList, ImagePathSet