update export csv error

visual-layer · Jun 15, 2023 · e890cee · e890cee
1 parent 3703058
commit e890cee
Show file tree

Hide file tree

Showing 2 changed files with 223 additions and 410 deletions.
diff --git a/notebooks/dataset-test.ipynb b/notebooks/dataset-test.ipynb
diff --git a/visuallayer/datasets/zoo/vl_oxford_iiit.py b/visuallayer/datasets/zoo/vl_oxford_iiit.py
@@ -43,7 +43,7 @@ def export(
                 print(
                     f"Exporting {variation.upper()} dataset into {output_format} dataset."
                 )
-                dataset = CleanTorchvisionOxfordIIITPet(root=self.root, split=split)
+                dataset = CleanTorchvisionOxfordIIITPet(root=self.root, split=split, exclude_csv=self.filelist_csv_url)
                 samples = {"Image": dataset._images, "Label": dataset._labels}
                 df = pd.DataFrame(samples)
                 return df