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))
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