Merge remote-tracking branch 'origin/master'

Selmaan · Apr 23, 2020 · c4f054b · c4f054b
2 parents d3f17ca + 94b8390
commit c4f054b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepposekit/io/BaseGenerator.py b/deepposekit/io/BaseGenerator.py
@@ -74,7 +74,7 @@ def get_images(self, indexes):
         Takes a list or array of indexes corresponding
         to image-keypoint pairs in the dataset.
         Returns a numpy array of images with the shape:
-        (1, height, width, n_channels)
+        (n_samples, height, width, n_channels)
         """
         raise NotImplementedError()
 
@@ -83,7 +83,7 @@ def get_keypoints(self, indexes):
         Takes a list or array of indexes corresponding to
         image-keypoint pairs in the dataset.
         Returns a numpy array of keypoints with the shape:
-        (1, n_keypoints, 2), where 2 is the x,y coordinates
+        (n_samples, n_keypoints, 2), where 2 is the x,y coordinates
         """
         raise NotImplementedError()