[Dataset] Add VOCSuperpixels dataset in LRGB (#6389)

dmlc · Oct 8, 2023 · 8edcad2 · 8edcad2
1 parent 403dba6
commit 8edcad2
Show file tree

Hide file tree

Showing 4 changed files with 331 additions and 6 deletions.
diff --git a/docs/source/api/python/dgl.data.rst b/docs/source/api/python/dgl.data.rst
@@ -70,6 +70,7 @@ Datasets for node classification/regression tasks
     MovieLensDataset
     PeptidesStructuralDataset
     PeptidesFunctionalDataset
+    VOCSuperpixelsDataset
 
 Edge Prediction Datasets
 ---------------------------------------

diff --git a/python/dgl/data/__init__.py b/python/dgl/data/__init__.py
@@ -75,7 +75,11 @@
 # Exception handling was added to prevent crashes for users who are using other
 # datasets.
 try:
-    from .lrgb import PeptidesFunctionalDataset, PeptidesStructuralDataset
+    from .lrgb import (
+        PeptidesFunctionalDataset,
+        PeptidesStructuralDataset,
+        VOCSuperpixelsDataset,
+    )
 except ImportError:
     pass
 from .pattern import PATTERNDataset