python+opencv2快速找出一组点包围的区域中的所有点

使用场景概述

假设你手头有一组点的坐标,你想找出这组坐标包围构成的多边形区域中的所有坐标点,即从边界生成分割掩模,也即cv2.findContours的反向操作。

先说这个需求的出发点,假设有一个我们有以下如图所示的AI自动分割的结果,蓝色的边界线是根据分割的二值掩模使用cv2.findContours确定并显示出来的。请添加图片描述
直观的一种修改模式(非AI辅助的情况下),人手拖拽边界点即可扩大或缩小边界。

方法介绍和代码分享

由于边界包含了过多且密集的点,使得在后续判定修改的具体是哪个点难以进行判定。为解决这一问题,我们先采用对边界进行均匀下采样,即相隔一定的样本数进行采样。

首先提取边界:

img = cv2.imread(image_path)
label = np.asarray(cv2.imread(label_path,0))

contours, _ = cv2.findContours(label.astype(np.uint8), cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
contours_lens = [len(c) for c in contours]
cnts = contours[np.argmax(contours_lens)]

如果不使用最后一句进行筛选,画出来发现其实contour不止是最边界的一条。

假设选取其中的25个点作为采样点,这25个采样点之间相隔相同的点数:

interval = cnts.shape[0] // 25
points = [cnts[i,:] for i in range(0,cnts.shape[0],interval)]
points = np.asarray(points).squeeze()

第一句是为了找出采样间隔,将总点数除以采样的样本数。
绘制出采样点

for point in points:
        radius = 5
        color=(0,255,0)
        thickness = 2
        cv2.circle(img,(point[0],point[1]),radius,color,thickness)

在这里插入图片描述现在假设我们改动了其中一个点:
在这里插入图片描述
现在有一个点飘出去了,那我们想从这组边界点推出现在的二值掩膜,其实就是找到这组边界包围内的所有点,应该怎么做呢?
下面的代码可以实现,且速度极快:

points = points.reshape(-1,1,2).astype(np.int32)
matrix =np.zeros((img.shape[0],img.shape[1]),dtype=np.int32)
cv2.drawContours(matrix,[points],-1,(1),thickness=-1)
list_of_points_indices=np.nonzero(matrix)

mask = np.zeros_like(label)
mask[list_of_points_indices] =1

效果如下:在这里插入图片描述

  • 4
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值