doc

torch_geometric/datasets/pascal.py

@@ -24,7 +24,7 @@ class PascalVOCKeypoints(InMemoryDataset):
     r"""The Pascal VOC 2011 dataset with Berkely annotations of keypoints from
     the `"Poselets: Body Part Detectors Trained Using 3D Human Pose
     Annotations" <https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/
-    human/ poselets_iccv09.pdf>`_ paper, containing 3 to 23 keypoints per
+    human/ poselets_iccv09.pdf>`_ paper, containing 0 to 23 keypoints per
     example over 20 categories.
     The dataset is pre-filtered to exclude difficult, occluded and truncated
     objects.

torch_geometric/datasets/pascal_pf.py

@@ -10,6 +10,32 @@
 
 
 class PascalPF(InMemoryDataset):
+    r"""The Pascal-PF dataset from the `"Proposal Flow"
+    <https://arxiv.org/abs/1511.05065>`_ paper, containing 4 to 16 keypoints
+    per example over 20 categories.
+
+    Args:
+        root (string): Root directory where the dataset should be saved.
+        category (string): The category of the images (one of
+            :obj:`"Aeroplane"`, :obj:`"Bicycle"`, :obj:`"Bird"`,
+            :obj:`"Boat"`, :obj:`"Bottle"`, :obj:`"Bus"`, :obj:`"Car"`,
+            :obj:`"Cat"`, :obj:`"Chair"`, :obj:`"Diningtable"`, :obj:`"Dog"`,
+            :obj:`"Horse"`, :obj:`"Motorbike"`, :obj:`"Person"`,
+            :obj:`"Pottedplant"`, :obj:`"Sheep"`, :obj:`"Sofa"`,
+            :obj:`"Train"`, :obj:`"TVMonitor"`)
+        transform (callable, optional): A function/transform that takes in an
+            :obj:`torch_geometric.data.Data` object and returns a transformed
+            version. The data object will be transformed before every access.
+            (default: :obj:`None`)
+        pre_transform (callable, optional): A function/transform that takes in
+            an :obj:`torch_geometric.data.Data` object and returns a
+            transformed version. The data object will be transformed before
+            being saved to disk. (default: :obj:`None`)
+        pre_filter (callable, optional): A function that takes in an
+            :obj:`torch_geometric.data.Data` object and returns a boolean
+            value, indicating whether the data object should be included in the
+            final dataset. (default: :obj:`None`)
+    """
     url = ('https://www.di.ens.fr/willow/research/proposalflow/dataset/'
            'PF-dataset-PASCAL.zip')