This would return NaN if total is zero

scikit-image · Jan 21, 2016 · 9689ed2 · 9689ed2
1 parent ce4bc59
commit 9689ed2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/skimage/feature/_hoghistogram.pyx b/skimage/feature/_hoghistogram.pyx
@@ -70,7 +70,9 @@ cdef float cell_hog(double[:, ::1] magnitude,
 
             total += magnitude[cell_row_index, cell_column_index]
 
-    return total / (cell_rows * cell_columns)
+    if total:
+        return total / (cell_rows * cell_columns)
+    return 0.
 
 def hog_histograms(double[:, ::1] gradient_columns,
                    double[:, ::1] gradient_rows,