基于 cumsum 从轮廓填充区域

Fill regions from contour based on cumsum

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

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

我在pytorch中进行了这个简单的操作(当然也可以使用tensorflow,或者python)

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

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

结果为: