File size: 342 Bytes
205a7af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
"""Entrypoint for torch hub."""
dependencies = ["torch", "torchvision", "opencv-python", "kornia", "matplotlib"]
from geocalib import GeoCalib
def model(*args, **kwargs):
"""Pre-trained Geocalib model.
Args:
weights (str): trained variant, "pinhole" (default) or "distorted".
"""
return GeoCalib(*args, **kwargs)
|