EndoGaussian-4D / models /__init__.py
mnunziant's picture
Add models package init
401f030 verified
Raw
History Blame Contribute Delete
282 Bytes
"""
EndoGaussian-4D Models Package
Contains:
- trainer.py: EndoGaussianTrainer with HexPlane deformation
- metrics.py: PSNR, SSIM, LPIPS, depth metrics
"""
from .metrics import compute_psnr, compute_ssim, LPIPSMetric, dssim_loss, compute_depth_metrics, MetricsAccumulator