From 354315502bdd922587407b1e049ff599c38cbd11 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 01:14:44 +0100 Subject: [PATCH] Modernize data loader, remove read image helper --- embedding_converter/src/data_loader.py | 2 +- embedding_converter/src/types.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/embedding_converter/src/data_loader.py b/embedding_converter/src/data_loader.py index 11f5a51..738abbc 100644 --- a/embedding_converter/src/data_loader.py +++ b/embedding_converter/src/data_loader.py @@ -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]): diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index a40d4bb..1a01548 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -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]