File size: 725 Bytes
6ffd3f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | # SPDX-License-Identifier: Apache-2.0
"""SuperPoint keypoint detection + description on a Tenstorrent Blackhole chip (tt-nn).
from tt_superpoint import SuperPoint
with SuperPoint.from_pretrained(device_id=0) as model:
out = model("image.jpg") # or a PIL image / numpy array / torch tensor
print(out.keypoints.shape, out.scores[:5], out.descriptors.shape)
See PYTHON.md for the full API reference.
"""
from .inputs import to_plane
from .model import DEFAULT_MODEL_ID, DEFAULT_REVISION, INPUT_SIZE, SuperPoint, SuperPointOutput
__version__ = "0.1.0"
__all__ = ["SuperPoint", "SuperPointOutput", "to_plane", "DEFAULT_MODEL_ID", "DEFAULT_REVISION", "INPUT_SIZE",
"__version__"]
|