mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-07-28 16:08:50 +02:00
Modernize data loader, remove read image helper
This commit is contained in:
@@ -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