fix: reshape images to match in *shannon_cross_entropy*

With this fix, we can now safely compare two images with different shapes.
ruancomelli · Jun 23, 2021 · 95f9855 · 95f9855
1 parent 666db0d
commit 95f9855
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/boiling_learning/preprocessing/image.py b/boiling_learning/preprocessing/image.py
@@ -295,6 +295,8 @@ def retained_variance(ref: np.ndarray, image: np.ndarray) -> float:
 def shannon_cross_entropy(
     ref: np.ndarray, image: np.ndarray, nbins: int = 100
 ) -> float:
+    ref, image = reshape_to_largest(ref, image)
+
     ref_histogram, _ = histogram(ref, nbins=nbins)
     img_histogram, _ = histogram(image, nbins=nbins)