Function open_folder_images

utils/poster_eval_utils.py:723–781

Opens all PNG images in folder_path named '{paper_name}-{index}.png', starting from index=1 up to the first missing, and returns them either as file-paths (if return_path=True) or as PIL.Image objects. If img_format!='png', each PNG is downsampled to fit within max_size (pr

(
    folder_path,
    paper_name,
    return_path=False,
    format='png',
    max_size=(700, 700),
    quality=80
)

    return question_only, answers, aspects

def open_folder_images(
    folder_path,
    paper_name,
    return_path=False,
    format='png',
    max_size=(700, 700),
    quality=80
):
    """
    Opens all PNG images in folder_path named '{paper_name}-{index}.png',
    starting from index=1 up to the first missing, and returns them
    either as file-paths (if return_path=True) or as PIL.Image objects.
    
    If img_format!='png', each PNG is downsampled to fit within max_size
    (preserving aspect ratio), converted to RGB, and saved into an
    in-memory JPEG with the given quality, optimize and progressive flags.
    """
    images = []
    index = 1

    while True:
        png_name = f"{paper_name}-{index}.png"
        path = os.path.join(folder_path, png_name)
        if not os.path.isfile(path):
            break

        if format == 'png':
            if return_path:
                images.append(path)
            else:
                images.append(Image.open(path))
        else:
            # 1) Load and downsample
            with Image.open(path) as im:
                thumb = im.copy()
                thumb.thumbnail(max_size, resample=Image.LANCZOS)

                # 2) Convert & compress to JPEG in-memory
                rgb = thumb.convert("RGB")
                buf = BytesIO()
                rgb.save(
                    buf,
                    format="JPEG",
                    quality=quality,        # e.g. 80–90
                    optimize=True,          # extra pass to strip redundant data
                    progressive=True        # for incremental rendering
                )
                buf.seek(0)

                if return_path:
                    # we return a tuple of (fake-jpg-filename, buffer)
                    jpg_name = png_name.rsplit('.', 1)[0] + '.jpg'
                    images.append((jpg_name, buf))
                else:
                    images.append(Image.open(buf))

        index += 1