Update create_test_dataset

erfect2020 · Nov 17, 2024 · 8333af1 · 8333af1
1 parent 5721584
commit 8333af1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/create_test_dataset b/data/create_test_dataset
@@ -55,4 +55,4 @@ class TestDataset(Dataset):
         gt_img = torch.tensor(gt_img / 255.0).float().permute(2, 0, 1)
         b_img = torch.tensor(b_img / 255.).float().permute(2, 0, 1)
 
-        return b_img, gt_img, os.path.basename(c_img_name).split('.')
+        return gt_img, b_img, os.path.basename(c_img_name).split('.')