From 49c509bcb30f978b1052b96811325c7188b876ab Mon Sep 17 00:00:00 2001
From: mattminder <myfiles@vpn-251-030.epfl.ch>
Date: Wed, 13 May 2020 10:06:12 +0200
Subject: [PATCH] type error: boolean array fix
---
unet/segment.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/unet/segment.py b/unet/segment.py
index d304893..43fde00 100644
--- a/unet/segment.py
+++ b/unet/segment.py
@@ -39,10 +39,10 @@ def cell_merge(wsh, pred):
wshshape=wsh.shape
# masks for the original cells
- objs = np.zeros((wsh.max()+1,wshshape[0],wshshape[1]))
+ objs = np.zeros((wsh.max()+1,wshshape[0],wshshape[1]), dtype=bool)
# masks for dilated cells
- dil_objs = np.zeros((wsh.max()+1,wshshape[0],wshshape[1]))
+ dil_objs = np.zeros((wsh.max()+1,wshshape[0],wshshape[1]), dtype=bool)
# bounding box coordinates
obj_coords = np.zeros((wsh.max()+1,4))
--
GitLab