From 1a66169b3737d6951a2c2294d6b634eaa3c3b0c1 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Wed, 5 Feb 2025 21:33:33 +0100 Subject: [PATCH] updated keys --- src/aiunn/finetune.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py index 7956384..cb56565 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -34,8 +34,8 @@ class aiuNNDataset(torch.utils.data.Dataset): def __getitem__(self, idx): # Get the byte strings - low_res_bytes = self.df.iloc[idx]['low_res'] - high_res_bytes = self.df.iloc[idx]['high_res'] + low_res_bytes = self.df.iloc[idx]['low_ress'] + high_res_bytes = self.df.iloc[idx]['high_ress'] # Convert bytes to numpy arrays low_res = cv2.imdecode(np.frombuffer(low_res_bytes, np.uint8), -1)