def extract_rgb_images()

in quality_comparison/extract_mp3d_real.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

def extract_rgb_images(pano_path: str, save_prefix: str) -> None:
    pano = imageio.imread(pano_path)
    # Convert pano to images
    list_of_images = []
    locs = np.linspace(0, 1.0, num=10)[:-1]
    np.random.shuffle(locs)
    for loc in locs[:NUM_IMAGES_PER_PANO]:
        pimg = e2p(pano, (FOV, FOV), (loc - 0.5) * 360.0, 0.0, (HEIGHT, WIDTH))
        list_of_images.append(pimg)
    # Save images
    for idx, image in enumerate(list_of_images):
        path = save_prefix + f"_img_{idx:03d}.jpg"
        imageio.imwrite(path, image)