transnormal2/__init__.py

"""TransNormal-2: one-step rectified-flow surface normal estimation."""

from .gsm import GSM
from .lcm import LocalContinuityModule
from .pipeline import TransNormal2Pipeline
from .utils import load_image, save_normal_map

__all__ = [
    "GSM",
    "LocalContinuityModule",
    "TransNormal2Pipeline",
    "load_image",
    "save_normal_map",
]

__version__ = "1.0.0"