Currently the mask weights are set to 1 ([source code](https://github.com/datajoint/element-zstack/blob/ebae639022473b06333eeb02c666c8e10213c2de/element_zstack/volume.py#L246)). ```python mask_weights = np.full_like(mask_zpix, 1) ``` Update the line above to use the mask weights returned from Cellpose (i.e. [flows](https://cellpose.readthedocs.io/en/latest/outputs.html)). Previously discussed [here](https://github.com/datajoint/element-zstack/pull/3#discussion_r1142184292).