mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-06-25 07:59:55 +02:00
Modernize data loader, remove read image helper
This commit is contained in:
@@ -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]):
|
||||
|
||||
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user