ENH adding new tasks (MRI, demoisaicing)

.gitignore

@@ -18,3 +18,4 @@ benchopt.ini
 coverage.xml
 
 tmp
+data/

datasets/bsd500_cbsd68.py

@@ -90,17 +90,22 @@ def get_data(self):
         ])
 
         path = get_data_path("BSD500")
-        train_dataset = dinv.datasets.BSDS500(
+        bsd500_dataset = dinv.datasets.BSDS500(
             path, download=True, transform=transform
         )
+        train_dataset = HuggingFaceTorchDataset(
+            bsd500_dataset,
+            key=...,
+            physics=physics,
+            device=device,
+            transform=transforms.Resize((self.img_size, self.img_size))
+        )
 
         dataset_cbsd68 = load_dataset("deepinv/CBSD68")
         test_dataset = HuggingFaceTorchDataset(
             dataset_cbsd68["train"],
             key="png",
             physics=physics,
-            save_dir=get_data_path("bsd500_cbsd68"),
-            dataset_filename=self.task,
             device=device
         )
 

datasets/bsd500_imnet100.py

@@ -90,9 +90,16 @@ def get_data(self):
         ])
 
         path = get_data_path("BSD500")
-        train_dataset = dinv.datasets.BSDS500(
+        bsd500_dataset = dinv.datasets.BSDS500(
             path, download=True, transform=transform
         )
+        train_dataset = HuggingFaceTorchDataset(
+            bsd500_dataset,
+            key=...,
+            physics=physics,
+            device=device,
+            transform=transforms.Resize((self.img_size, self.img_size))
+        )
 
         dataset_miniImnet100 = load_dataset("mterris/miniImnet100")
         test_dataset = HuggingFaceTorchDataset(