Merge pull request #605 from aleju/fix_jigsaw_error

Fix error message in Jigsaw
aleju · Feb 2, 2020 · d3fac0a · d3fac0a
2 parents 35e5f47 + 62e9a40
commit d3fac0a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/imgaug/augmenters/geometric.py b/imgaug/augmenters/geometric.py
@@ -5921,8 +5921,10 @@ def _augment_batch_(self, batch, random_state, parents, hooks):
                                    "line_strings"]])
         if has_other_cbaoi:
             raise NotImplementedError(
-                "Jigsaw currently only supports augmentation of images "
-                "and keypoints.")
+                "Jigsaw currently only supports augmentation of images, "
+                "heatmaps, segmentation maps and keypoints. "
+                "Explicitly not supported are: bounding boxes, polygons "
+                "and line strings.")
 
         # We don't crop back to the original size, partly because it is
         # rather cumbersome to implement, partly because the padded