import random


def get_starts(N, M, batch_size, option):
    """
    Given the number of elevations(N), azimuths(M), batch size and the option (different types of starts),
    this function returns the start indices for the batch.
    start_idx: list of [start_elev, start_azim] for each panorama in the batch
    """
    if option == 0:
        start_idx = [[random.randint(0, N-1), random.randint(0, M-1)] for i in range(batch_size)]
    else:
        start_idx = [[N//2, M//2-1] for i in range(batch_size)]

    return start_idx