From b4822ee64ada88b3c3d9d55de91a24f20fefe0ab Mon Sep 17 00:00:00 2001
From: mattminder <myfiles@vpn-251-030.epfl.ch>
Date: Wed, 13 May 2020 10:01:49 +0200
Subject: [PATCH] bugfix: min/max calculation
---
unet/segment.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/unet/segment.py b/unet/segment.py
index 2ca2670..d304893 100644
--- a/unet/segment.py
+++ b/unet/segment.py
@@ -146,6 +146,5 @@ def get_bounding_box(im):
"""Returns bounding box of object in boolean image"""
coords = np.where(im)
- min0, min1 = coords.min(axis=0)
- max0, max1 = coords.max(axis=0)
- return np.array([min0, max0, min1, max1])
+ return np.array([np.min(coords[0]), np.max(coords[0]),
+ np.min(coords[1]), np.max(coords[1])])
--
GitLab