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

15 lines
383 B
Python

from typing import Any, Tuple, List, Dict, Optional
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]
Embedding = NDArray[Any]
VisionFrame = NDArray[Any]