Merge pull request #34 from mpelchat04/binary_inference

Fix issue with binary scripted model
NRCan · Dec 4, 2024 · 058de06 · 058de06
2 parents e7ceb9a + da4715b
commit 058de06
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/geo_inference/geo_dask.py b/geo_inference/geo_dask.py
@@ -298,7 +298,6 @@ def sum_overlapped_chunks(
                     where=full_array[-1, :, :] != 0,
                 )
                 if final_result.shape[0] == 1:
-                    final_result = expit(final_result)
                     final_result = (
                         np.where(final_result > prediction_threshold, 1, 0).squeeze(0).astype(np.uint8)
                     )