Broadcasting is work except in some cases that appear to coincide with...

Broadcasting is work except in some cases that appear to coincide with truncation of ROI in patch bounds

model_server/base/roiset.py

@@ -229,8 +229,14 @@ class RoiSet(object):
         zmask = np.zeros((*acc_raw.hw, 1, acc_raw.nz), dtype=bool)
 
         def _label_obj(r):
-            sl = np.s_[r.ebb_y0:r.ebb_y1, r.ebb_x0:r.ebb_x1, :, r.zi:r.zi + 1]
-            zmask[sl] = patch_masks[r.label]
+            # sl = np.s_[r.ebb_y0:r.ebb_y1, r.ebb_x0:r.ebb_x1, :, r.zi:r.zi + 1]
+            sl = np.s_[r.y0:r.y1, r.x0:r.x1, :, r.zi:r.zi + 1]
+            rsl = np.s_[r.rel_y0:r.rel_y1, r.rel_x0:r.rel_x1, :, :]
+            try:
+                zmask[sl] = patch_masks[r.label][rsl]
+                print(f'ebb (h, w) = {r.ebb_h, r.ebb_w}')
+            except ValueError as ve:
+                print(ve)
 
         df.apply(lambda x: _label_obj(x), axis=1)
         obj_ids = _get_label_ids(zmask, allow_3d=True, connect_3d=False)