Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

develop #4

Merged
Fabel merged 103 commits from develop into main 2025-03-01 21:47:17 +00:00
Conversation 0 Commits 103 Files Changed 13 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 72a5959bc1 - Show all commits

4
src/aiunn/finetune.py
View File

@ -34,8 +34,8 @@ class aiuNNDataset(torch.utils.data.Dataset):
def __getitem__(self, idx): def __getitem__(self, idx):
# Get the byte strings # Get the byte strings
low_res_bytes = self.df.iloc[idx]['low_ress'] low_res_bytes = self.df.iloc[idx]['image_512']
high_res_bytes = self.df.iloc[idx]['high_ress'] high_res_bytes = self.df.iloc[idx]['image_1024']
# Convert bytes to numpy arrays # Convert bytes to numpy arrays
low_res = cv2.imdecode(np.frombuffer(low_res_bytes, np.uint8), -1) low_res = cv2.imdecode(np.frombuffer(low_res_bytes, np.uint8), -1)