基于 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)
结果为:
我得到一张用于场景分类的图像:
并且使用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)
结果为: