Python 基于累积和的等高线填充区域

我得到了一张用于场景分类的图像：

使用cumsum，我想把它的三个部分进行分段

我在pytorch中执行这个简单的操作（当然也可以使用tensorflow或python）

您可以使用它在两个方向执行

cumsum

：

mask=（src_img.cumsum（dim=0）>0）+2*torch.flipud（torch.flipud（src_img.cumsum）（dim=0）>0）

导致：

---

难道你不能自下而上进行累积求和，然后将这两个结果进行计算吗？@BillKid你的问题中似乎没有图像了！