mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-04-19 15:56:37 +02:00
Modernize data loader, remove read image helper
This commit is contained in:
@@ -3,8 +3,8 @@ import random
|
||||
|
||||
import cv2
|
||||
import torch
|
||||
import transforms from torchvision
|
||||
from torch.utils.data import Dataset
|
||||
from torchvision import transforms
|
||||
|
||||
from .types import Batch, Paths
|
||||
|
||||
|
||||
@@ -5,8 +5,8 @@ from typing import Tuple
|
||||
|
||||
import cv2
|
||||
import torch
|
||||
import transforms from torchvision
|
||||
from torch.utils.data import TensorDataset
|
||||
from torchvision import transforms
|
||||
|
||||
from .types import Batch, ImagePathList, ImagePathSet
|
||||
|
||||
|
||||
Reference in New Issue
Block a user