Efficient-Large-Model/pi3x-intrinsics-estimator

Pi3X intrinsics estimator

A remote modular block that estimates camera intrinsics [fx, fy, cx, cy] for a single image using Pi3X, for use with SanaWMPipeline.

import torch
from diffusers import ModularPipeline
from diffusers.utils import load_image

estimator = ModularPipeline.from_pretrained(
    "Efficient-Large-Model/pi3x-intrinsics-estimator", trust_remote_code=True
)
estimator.load_components(dtype=torch.bfloat16)
intrinsics = estimator(image=load_image("input.png"), output="intrinsics")

Requires pip install pi3-vision torchvision.