viz_dataset.py [39:47]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    longs = coords[:, 0].detach().cpu().numpy()
    lats = coords[:, 1].detach().cpu().numpy()

    if MAPS[dataset_name]:
        map_img = plt.imread(MAPS[dataset_name])
        fig, ax = plt.subplots(figsize=(FIGSIZE, FIGSIZE * map_img.shape[0] / map_img.shape[1]))
        ax.imshow(map_img, zorder=0, extent=BBOXES[dataset_name])
    else:
        fig, ax = plt.subplots(figsize=(FIGSIZE, FIGSIZE))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



viz_dataset.py [63:71]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    longs = coords[:, 0].detach().cpu().numpy()
    lats = coords[:, 1].detach().cpu().numpy()

    if MAPS[dataset_name]:
        map_img = plt.imread(MAPS[dataset_name])
        fig, ax = plt.subplots(figsize=(FIGSIZE, FIGSIZE * map_img.shape[0] / map_img.shape[1]))
        ax.imshow(map_img, zorder=0, extent=BBOXES[dataset_name])
    else:
        fig, ax = plt.subplots(figsize=(FIGSIZE, FIGSIZE))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -