Modernize data loader, remove read image helper

This commit is contained in:
henryruhs
2025-02-21 01:14:44 +01:00
parent 254bc17c98
commit 354315502b
2 changed files with 1 additions and 2 deletions
+1 -1
View File
@@ -6,7 +6,7 @@ import torch
from torch.utils.data import Dataset
from torchvision import transforms
from .types import Batch, Paths
from .types import Batch
class DataLoaderRecognition(Dataset[torch.Tensor]):
-1
View File
@@ -3,7 +3,6 @@ from typing import Any, List, TypeAlias
from numpy.typing import NDArray
from torch import Tensor
Paths : TypeAlias = List[str]
Batch : TypeAlias = Tensor
Embedding : TypeAlias = Tensor
VisionFrame : TypeAlias = NDArray[Any]