首页 > 后端开发 > C++ > 如何使用 Python 中的位图和矢量方法有效地检测和勾画一组分散的 2D 点中的孔?

如何使用 Python 中的位图和矢量方法有效地检测和勾画一组分散的 2D 点中的孔?

Susan Sarandon
发布: 2025-01-18 07:26:09
原创
457 人浏览过

How can I efficiently detect and outline holes within a scattered set of 2D points using a bitmap and vector approach in Python?

此 Python 代码提供了类似于上一个答案中描述的位图矢量方法的实现。它通过计算位图中的数据密度、识别未使用的区域、分割输出并对结果进行多边形化来查找一组 2D 点中的孔。

import numpy as np
import cv2
import matplotlib.pyplot as plt

def find_holes(points, resolution=100):
    """Find holes in a set of 2D points.

    Args:
        points: A list of (x, y) tuples representing the points.
        resolution: The resolution of the bitmap to use. Higher resolution
            results in more accurate results, but is slower.

    Returns:
        A list of (x, y) tuples representing the vertices of the holes.
    """
    # Create a bitmap of the points.
    bitmap = np.zeros((resolution, resolution), dtype=np.uint8)
    for point in points:
        x, y = point
        bitmap[int(y * resolution), int(x * resolution)] = 255

    # Compute data density in the bitmap.
    density = cv2.dilate(bitmap, np.ones((3, 3))) - cv2.erode(bitmap, np.ones((3, 3)))

    # Identify unused areas in the bitmap.
    unused_areas = np.where(density == 0)

    # Segment the unused areas.
    segmented_areas = cv2.watershed(density, np.zeros((resolution, resolution), dtype=np.int32), markers=unused_areas[0], mask=bitmap)

    # Polygonize the segmented areas.
    holes = []
    for i in range(1, np.max(segmented_areas) + 1):
        mask = np.zeros((resolution, resolution), dtype=np.uint8)
        mask[segmented_areas == i] = 255
        _, contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
        for contour in contours:
            hole = []
            for point in contour:
                x, y = point[0]
                hole.append((x / resolution, y / resolution))
            holes.append(hole)

    return holes


# Example usage

# Generate a set of random points.
points = [(np.random.rand(), np.random.rand()) for _ in range(1000)]

# Find the holes in the set of points.
holes = find_holes(points, resolution=50)

# Plot the points and the holes.
plt.scatter([x for (x, y) in points], [y for (x, y) in points], s=1, c='black')
for hole in holes:
    plt.plot([x for (x, y) in hole], [y for (x, y) in hole], c='red')
plt.show()
登录后复制

此代码使用 OpenCV 进行位图操作,使用 NumPy 进行数组操作操作,以及用于绘图的 matplotlib。它可以轻松修改以使用不同的数据类型和坐标系。

以上是如何使用 Python 中的位图和矢量方法有效地检测和勾画一组分散的 2D 点中的孔?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板