Skip to content
Snippets Groups Projects
Commit 49c509bc authored by mattminder's avatar mattminder
Browse files

type error: boolean array fix

parent b4822ee6
No related branches found
No related tags found
No related merge requests found
...@@ -39,10 +39,10 @@ def cell_merge(wsh, pred): ...@@ -39,10 +39,10 @@ def cell_merge(wsh, pred):
wshshape=wsh.shape wshshape=wsh.shape
# masks for the original cells # 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 # 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 # bounding box coordinates
obj_coords = np.zeros((wsh.max()+1,4)) obj_coords = np.zeros((wsh.max()+1,4))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment