Update app.py

app.py

@@ -269,7 +269,7 @@ def update(filepath, filepath_result, filepath_coordinates, z, c):
     return (filepath_show, display_boxes), (filepath_result_show, display_boxes)
 
 
-def filter_coordinates_alt(filepath_coordinates, z, image_shape=(512, 512, 10)):
+def filter_coordinates_alt(filepath_coordinates, z, image_shape=(512, 512)):
 
     depth = 3
     coordinates = np.loadtxt(filepath_coordinates, delimiter=",")
@@ -283,22 +283,30 @@ def filter_coordinates_alt(filepath_coordinates, z, image_shape=(512, 512, 10)):
     #rel_z = np.abs(coordinates[:, 0]-z)
     #rel_z = rel_z[:, np.newaxis]
 
-    markers = np.zeros((image_shape[0], image_shape[1], 2*depth-1))
+    boxes = np.zeros(image_shape)
     
     x_coord = tuple(coordinates[:,2])
     y_coord = tuple(coordinates[:,1])
-    rel_z_coord = tuple(coordinates[:,0]-z) 
-    print(rel_z_coord)
-    values = [1] * len(rel_z_coord )
+    size = tuple(4-np.abs(coordinates[:,0]-z))
+
+    for x, y, size in zip(x_coord, y_coord, size):
+        boxes = draw_box(markers, x, y, size)
+
 
     markers[x_coord, y_coord, rel_z_coord] = values
     
-    print(rel_z_coord)
-    
-    boxes = sc.ndimage.gaussian_filter(markers, sigma=(5,5, depth)) > 0.001 
+    return [(boxes,'nothing')]
 
-    print(boxes)
-    return [(boxes[:,:, depth],'nothing')]
+def draw_box(array, x, y, size):
+    x0 = max(x-size, 0)
+    y0 = max(y-size, 0)
+
+    x1 = min(x+size+1, array.shape[0]-1)
+    y1 = min(y+size+1, array.shape[1]-1)
+
+    array[x0:x1, y0:y1] = 1
+
+    return array
 
 def add_boxes_norm_path(filepath, boxes):
     img = imread(filepath)