replay_trajectory_classification.environments.get_track_segments_from_graph#

get_track_segments_from_graph(track_graph: Graph) ndarray[source]#

Returns a 2D array of node positions corresponding to each edge.

Parameters:

track_graph (networkx Graph)

Returns:

track_segments

Return type:

np.ndarray, shape (n_segments, n_nodes, n_space)