基于 cumsum 从轮廓填充区域

Question

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

并且使用 cumsum，我想分割它的三个部分。

我在pytorch中做这个简单的操作（当然也可以用tensorflow，或者python）

Answer 1

您可以使用torch.flipud 在两个方向上执行cumsum：

mask = (src_img.cumsum(dim=0) >0 ) + 2* torch.flipud(torch.flipud(src_img).cumsum(dim=0)>0)

结果：