Как преобразовать контуры, полученные из OpenCV, в полигоны файла SHP? - PullRequest
Новая
       36

Как преобразовать контуры, полученные из OpenCV, в полигоны файла SHP?

0 голосов
/ 01 апреля 2020

Я успешно извлек контуры из изображения маски, используя OpenCV: 

image = cv2.imread(file)

lower = np.array([240, 240, 240])
upper = np.array([255, 255, 255])
shape_mask = cv2.inRange(image, lower, upper)

contours, hierarchy = cv2.findContours(shape_mask.copy(), cv2.RETR_CCOMP,
                                       cv2.CHAIN_APPROX_SIMPLE)

Теперь, как мне go включить, чтобы отобразить эти контуры в список многоугольников (с отверстиями) с фигурными для экспорта их как файл SHP?

Здесь дан частичный ответ: Как преобразовать NumPy массивы, полученные из cv2.findContours, в полигоны Shapely?

Однако это игнорирует случай наличия отверстий внутри многоугольников. Как мне go получить все полигоны?

1 Ответ

0 голосов
/ 01 апреля 2020

Найден ответ: https://michhar.github.io/masks_to_polygons_and_back/

Помощник для создания мультиполигонов из маскированного изображения в виде numpy массива: 

def mask_to_polygons(mask, epsilon=10., min_area=10.):
    """Convert a mask ndarray (binarized image) to Multipolygons"""
    # first, find contours with cv2: it's much faster than shapely
    image, contours, hierarchy = cv2.findContours(mask,
                                  cv2.RETR_CCOMP,
                                  cv2.CHAIN_APPROX_NONE)
    if not contours:
        return MultiPolygon()
    # now messy stuff to associate parent and child contours
    cnt_children = defaultdict(list)
    child_contours = set()
    assert hierarchy.shape[0] == 1
    # http://docs.opencv.org/3.1.0/d9/d8b/tutorial_py_contours_hierarchy.html
    for idx, (_, _, _, parent_idx) in enumerate(hierarchy[0]):
        if parent_idx != -1:
            child_contours.add(idx)
            cnt_children[parent_idx].append(contours[idx])
    # create actual polygons filtering by area (removes artifacts)
    all_polygons = []
    for idx, cnt in enumerate(contours):
        if idx not in child_contours and cv2.contourArea(cnt) >= min_area:
            assert cnt.shape[1] == 1
            poly = Polygon(
                shell=cnt[:, 0, :],
                holes=[c[:, 0, :] for c in cnt_children.get(idx, [])
                       if cv2.contourArea(c) >= min_area])
            all_polygons.append(poly)
    all_polygons = MultiPolygon(all_polygons)

    return all_polygons

Кредит для создания этих помощников functions:

Первоначальным источником этих помощников был пост Kaggle Константина Лопухина здесь - вам необходимо войти в Kaggle, чтобы увидеть его

...