Python make_graph_segm_connect_grid3d_conn6の例

コード例 #1

ファイル: test_superpixels.py プロジェクト: rjbashar/pyImSegm

    def test_segm_connect(self):
        logging.debug(self.seg2d)
        vertices, edges = seg_spx.make_graph_segm_connect_grid2d_conn4(self.seg2d)
        logging.debug('vertices: {} -> edges: {}'.format(vertices, edges))

        logging.debug(self.seg3d)
        vertices, edges = seg_spx.make_graph_segm_connect_grid3d_conn6(self.seg3d)
        logging.debug('vertices: {} -> edges: {}'.format(vertices, edges))

コード例 #2

def get_vertexes_edges(segments):
    """ wrapper - get list of vertexes edges for 2D / 3D images

    :param ndarray segments:
    :return:
    """
    if segments.ndim == 3:
        vertices, edges = make_graph_segm_connect_grid3d_conn6(segments)
    elif segments.ndim == 2:
        vertices, edges = make_graph_segm_connect_grid2d_conn4(segments)
    else:
        return None, None
    return vertices, edges