From 254bc17c989031ef579319f2a061e63e92ccaf8d Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Fri, 21 Feb 2025 01:13:37 +0100
Subject: [PATCH] Modernize data loader, remove read image helper

---
 embedding_converter/src/data_loader.py | 2 +-
 face_swapper/src/data_loader.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/embedding_converter/src/data_loader.py b/embedding_converter/src/data_loader.py
index 5585c5e..11f5a51 100644
--- a/embedding_converter/src/data_loader.py
+++ b/embedding_converter/src/data_loader.py
@@ -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
 
diff --git a/face_swapper/src/data_loader.py b/face_swapper/src/data_loader.py
index c562b8a..81f73a3 100644
--- a/face_swapper/src/data_loader.py
+++ b/face_swapper/src/data_loader.py
@@ -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