From 650551c19bf554468cf6ab1efa424778f2f8a906 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 14 Feb 2025 16:19:14 +0100 Subject: [PATCH] Simplify Batch type --- embedding_converter/src/types.py | 5 ++--- face_swapper/src/types.py | 4 ++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index 9221412..3a71965 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -3,11 +3,10 @@ from typing import Any, Tuple, TypeAlias from numpy.typing import NDArray from torch import Tensor +Batch : TypeAlias = Tuple[Tensor, Tensor] + Embedding : TypeAlias = NDArray[Any] EmbeddingDataset : TypeAlias = NDArray[Embedding] FaceLandmark5 : TypeAlias = NDArray[Any] VisionFrame : TypeAlias = NDArray[Any] -VisionTensor : TypeAlias = Tensor - -Batch : TypeAlias = Tuple[VisionTensor, VisionTensor] diff --git a/face_swapper/src/types.py b/face_swapper/src/types.py index b5551d4..3a7bbed 100644 --- a/face_swapper/src/types.py +++ b/face_swapper/src/types.py @@ -5,6 +5,8 @@ from numpy.typing import NDArray from torch import Tensor from torch.nn import Module +Batch : TypeAlias = Tuple[Tensor, Tensor, Tensor] + ImagePathList : TypeAlias = List[str] ImagePathSet : TypeAlias = Dict[str, ImagePathList] @@ -22,8 +24,6 @@ VisionFrame : TypeAlias = NDArray[Any] LossTensor : TypeAlias = Tensor VisionTensor : TypeAlias = Tensor -Batch : TypeAlias = Tuple[VisionTensor, VisionTensor, Tensor] - GeneratorLossSet : TypeAlias = Dict[str, Tensor] DiscriminatorLossSet : TypeAlias = Dict[str, Tensor]