如何约束实例分割的预测范围?

最近在做基于神经网络的医疗图像的器官分割预测,经常会出现预测结果包含了其他器官或非器官区域(比如超出人体皮肤),请问如何在分割的时候加入预测结果的范围约束,或者是否有相应的后处理方法?

这个我觉得可以直接惩罚,拉回到最近的范围边界即可!