Merge remote-tracking branch 'origin/master'

Author: jgraving

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()