Fixed jet() and heatmap() so they work on empty images.

89d6055f · Davis King · 97dc30bc · 89d6055f
Commit 89d6055f authored Jul 14, 2015 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

colormaps.h dlib/image_transforms/colormaps.h +8 -2

No files found.
--- a/dlib/image_transforms/colormaps.h
+++ b/dlib/image_transforms/colormaps.h
@@ -145,7 +145,10 @@ namespace dlib
    )
    {
        typedef op_heatmap<image_type> op;
-        return matrix_op<op>(op(img,max(mat(img)),min(mat(img))));
+        if (img.size() != 0)
+            return matrix_op<op>(op(img,max(mat(img)),min(mat(img))));
+        else
+            return matrix_op<op>(op(img,0,0));
    }

 // ----------------------------------------------------------------------------------------
@@ -252,7 +255,10 @@ namespace dlib
    )
    {
        typedef op_jet<image_type> op;
-        return matrix_op<op>(op(img,max(mat(img)),min(mat(img))));
+        if (img.size() != 0)
+            return matrix_op<op>(op(img,max(mat(img)),min(mat(img))));
+        else
+            return matrix_op<op>(op(img,0,0));
    }

 // ----------------------------------------------------------------------------------------