normalize

segment/seg.py

@@ -44,9 +44,13 @@ def segment_bf(well, thr=0.2, smooth=10, erode=10, fill=True, plot=False):
     labels, n_labels = multiwell.label(regions)
     print(f'{n_labels} region{" s"[n_labels>1]}')
     if plot:
-        fig, ax = multiwell.plt.subplots(1,2)
-        ax[0].imshow(sm, cmap='gray')
-        ax[1].imshow(labels)
+        fig, ax = multiwell.plt.subplots(1,3)
+        ax[0].imshow(well, cmap='gray')
+        ax[0].set_title('input')
+        ax[1].imshow(sm, cmap='gray')
+        ax[1].set_title('grad')
+        ax[2].imshow(labels)
+        ax[2].set_title(f'{n_labels} labels')
         multiwell.plt.show()
         
     return labels
\ No newline at end of file