Files
facefusion-labs/face_swapper/src/typing.py
T
2025-03-11 14:43:03 +01:00

16 lines
394 B
Python

from typing import Any, Dict, List, Tuple
from numpy.typing import NDArray
from torch import Tensor
from torch.utils.data import DataLoader
Batch = Tuple[Any, Any, Any]
Loader = DataLoader[Tuple[Tensor, ...]]
TargetAttributes = Tuple[Tensor, ...]
DiscriminatorOutputs = List[List[Tensor]]
LossDict = Dict[str, Tensor]
IDEmbedding = Tensor
Embedding = NDArray[Any]
VisionFrame = NDArray[Any]