Function read_image_frame

backend/frame_io.py:35–67

Read an image file (EXR or standard) as float32 RGB [0, 1]. Args: fpath: Absolute path to image file. gamma_correct_exr: If True, apply piecewise sRGB transfer function to EXR data (converts linear → sRGB for models expecting sRGB). Returns: float32 arra

(fpath: str, gamma_correct_exr: bool = False)

def read_image_frame(fpath: str, gamma_correct_exr: bool = False) -> np.ndarray | None:
    """Read an image file (EXR or standard) as float32 RGB [0, 1].

    Args:
        fpath: Absolute path to image file.
        gamma_correct_exr: If True, apply piecewise sRGB transfer function
            to EXR data (converts linear → sRGB for models expecting sRGB).

    Returns:
        float32 array [H, W, 3] in RGB order, or None if read fails.
    """
    is_exr = fpath.lower().endswith(".exr")

    if is_exr:
        img = cv2.imread(fpath, cv2.IMREAD_UNCHANGED)
        if img is None:
            logger.warning("Could not read frame: %s", fpath)
            return None
        # Strip alpha channel from BGRA EXR
        if img.ndim == 3 and img.shape[2] == 4:
            img = img[:, :, :3]
        img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        result = np.maximum(img_rgb, 0.0).astype(np.float32)
        if gamma_correct_exr:
            result = linear_to_srgb(result).astype(np.float32)
        return result
    else:
        img = cv2.imread(fpath)
        if img is None:
            logger.warning("Could not read frame: %s", fpath)
            return None
        img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        return img_rgb.astype(np.float32) / 255.0


def read_video_frame_at(