Fixed a potential divide by zero in draw_fhog()

793a8a51 · Davis King · c6d778ea · 793a8a51
Commit 793a8a51 authored May 16, 2014 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

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

No files found.
--- a/dlib/image_transforms/fhog.h
+++ b/dlib/image_transforms/fhog.h
@@ -847,7 +847,10 @@ namespace dlib
        }

        const double thresh = mean(himg) + 4*stddev(himg);
-        return matrix_cast<unsigned char>(upperbound(round(himg*255/thresh),255));
+        if (thresh != 0)
+            return matrix_cast<unsigned char>(upperbound(round(himg*255/thresh),255));
+        else
+            return matrix_cast<unsigned char>(himg);
    }

 // ----------------------------------------------------------------------------------------
@@ -927,7 +930,10 @@ namespace dlib
        }

        const double thresh = mean(himg) + 4*stddev(himg);
-        return matrix_cast<unsigned char>(upperbound(round(himg*255/thresh),255));
+        if (thresh != 0)
+            return matrix_cast<unsigned char>(upperbound(round(himg*255/thresh),255));
+        else
+            return matrix_cast<unsigned char>(himg);
    }

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