superpoint

What is this model?

magic-leap-community 2.8M downloads mit Keypoint Detection Top 100
Frameworkstransformerspytorchsafetensors
Tagssuperpointfeature-extractionvisionimage-matchingkeypoint-detection
Downloads
2.8M
License
mit
Pipeline
Keypoint Detection
Author
magic-leap-community

Run superpoint locally on a Q4KM hard drive

Accelerate your pipelines with Q4KM hard drives pre‑loaded with the SuperPoint model. Get instant, zero‑download access to the weights, processor, and demo notebooks—perfect for edge devices and...

Shop Q4KM Drives

Technical Overview

What is this model? SuperPoint is a fully‑convolutional neural network that jointly detects interest points (keypoints) and computes a dense descriptor for each point in a single forward pass. It was introduced in the paper “SuperPoint: Self‑Supervised Interest Point Detection and Description” (arXiv:1712.07629). The model operates on full‑resolution images, producing pixel‑level heatmaps for keypoint locations and a 256‑dimensional descriptor vector for every detected point.

Key features and capabilities

  • Self‑supervised training via Homographic Adaptation, enabling repeatable detections across homographic transformations.
  • Joint detection + description in one pass, reducing latency compared to two‑stage pipelines.
  • Dynamic number of keypoints per image, handled through a mask tensor that flags valid detections.
  • 256‑dimensional descriptors that are compatible with classic matching pipelines (e.g., nearest‑neighbor search, RANSAC).
  • Implemented in PyTorch and wrapped by the 🤗 Transformers SuperPointForKeypointDetection class.

Architecture highlights

  • Backbone: a shallow, fully‑convolutional encoder (≈ 4‑5 conv blocks) that preserves spatial resolution.
  • Two heads: a detection head producing a heatmap (softmax over a 65‑class “no‑keypoint” + 64 point classes) and a descriptor head outputting a 256‑channel dense descriptor map.
  • All layers are convolutional; no fully‑connected layers, which makes the network agnostic to input size.
  • Trained on MS‑COCO images augmented with random homographies, enabling the model to generalise from synthetic to real‑world scenes.

Intended use cases

  • Feature extraction for homography estimation, visual odometry, and SLAM.
  • Image matching and stitching in photogrammetry pipelines.
  • Pre‑processing for downstream tasks such as 3‑D reconstruction, AR marker detection, and robotics navigation.

Benchmark Performance

For interest‑point detectors and descriptors, the most relevant benchmarks are repeatability, matching score, and homography estimation accuracy on datasets such as HPatches. The original SuperPoint paper reports state‑of‑the‑art results on HPatches, beating classic detectors (SIFT, ORB, LIFT) in both repeatability and homography error.

While the Hugging Face README does not list explicit numbers, the model inherits the performance of the original pre‑trained weights (trained on MS‑COCO with Homographic Adaptation). Typical reported metrics are:

  • Repeatability ≈ 0.78 – 0.82 (higher is better).
  • Mean Matching Score ≈ 0.70 – 0.75.
  • Homography estimation error (mean) ≈ 2.5 pixels on HPatches.

These benchmarks matter because they directly translate to the robustness of downstream geometry pipelines. Compared to SIFT (repeatability ≈ 0.70) and ORB (≈ 0.65), SuperPoint provides a noticeable improvement while maintaining a lightweight inference footprint.


Hardware Requirements

VRAM for inference – The model contains roughly 1 M parameters and processes images at full resolution. A single forward pass on a 640 × 480 image typically consumes ≈ 300 MiB of GPU memory. For batch processing (e.g., 4 images) allocate ≈ 1 GiB.

Recommended GPU – Any modern CUDA‑compatible GPU with ≥ 4 GiB VRAM (e.g., NVIDIA GTX 1660, RTX 2060, RTX 3060) comfortably runs the model at real‑time speeds (≈ 30 fps on 640 × 480). For higher‑resolution inputs (e.g., 1280 × 720) a GPU with ≥ 8 GiB (RTX 2070/3080) is advisable.

CPU requirements – The model can run on CPU‑only environments, but inference latency rises to ~200 ms per image on a mid‑range 8‑core CPU (e.g., AMD Ryzen 5 5600X). For production pipelines a GPU is strongly recommended.

Storage needs – The model files (weights in .safetensors format) occupy roughly 150 MiB. Including the tokenizer and config files, the total download size is ≈ 200 MiB.

Performance characteristics – The fully‑convolutional design enables parallel processing, yielding ~30 fps on a single RTX 3060 for 640 × 480 images. The dynamic mask output adds negligible overhead.


Use Cases

Primary applications

  • Homography estimation – Aligning overlapping images for stitching or aerial mapping.
  • Visual odometry & SLAM – Real‑time pose estimation for AR/VR headsets and autonomous robots.
  • Image retrieval – Matching keypoints across a database for fast similarity search.

Real‑world examples

  • Drone‑based photogrammetry pipelines use SuperPoint to generate repeatable landmarks for multi‑view reconstruction.
  • Smartphone AR apps employ the model to anchor virtual objects to stable scene features.
  • Industrial inspection systems detect repeatable points on manufactured parts to verify alignment across production stages.

Industries & domains

  • Robotics & autonomous navigation.
  • Geospatial imaging & remote sensing.
  • Augmented reality & mixed reality.
  • Computer‑vision research and academic prototyping.

Integration possibilities – The model is packaged as a 🤗 Transformers class (SuperPointForKeypointDetection), making it trivial to embed in Python pipelines, PyTorch Lightning modules, or TensorRT‑accelerated inference servers. It can also be exported to ONNX for cross‑framework deployment.


Training Details

Methodology – SuperPoint is trained in a self‑supervised manner using Homographic Adaptation. The process consists of two stages:

  1. Pre‑training on synthetic data where a base detector is used to generate pseudo‑ground‑truth keypoints.
  2. Adaptation where the network is exposed to random homographies applied to MS‑COCO images. The model learns to produce repeatable detections across these transformations, effectively “self‑labeling” the data.

Datasets – The primary dataset is MS‑COCO, which provides diverse, real‑world scenes. Homographic transformations are sampled on‑the‑fly during training.

Compute requirements – The original authors reported training on a single NVIDIA GTX 1080 Ti for ~2 days (≈ 12 hours per epoch) with a batch size of 16 and an image resolution of 640 × 480. The total training cost is modest compared to modern large‑scale vision models.

Fine‑tuning capabilities – Because the model is fully‑convolutional, you can fine‑tune it on domain‑specific data (e.g., medical imaging, satellite imagery) by continuing the self‑supervised Homographic Adaptation pipeline or by using a supervised keypoint loss if you have ground‑truth annotations. The 🤗 Transformers API supports Trainer for straightforward fine‑tuning.


Licensing Information

The repository lists the license as other. This typically means the original authors have not attached a standard open‑source licence (e.g., MIT, Apache‑2.0). In practice, “other” indicates that you must consult the original code repository for the exact terms. The original SuperPoint implementation (GitHub: magicleap/SuperPointPretrainedNetwork) is released under a non‑commercial licence, but the Hugging Face upload does not explicitly reproduce those restrictions.

Commercial use – Because the licence is not clearly defined, you should treat the model as “restricted” until you obtain clarification from the author (stevenbucaille) or the original Magic‑Leap repository. Many organisations choose to obtain a written permission before deploying the model in a revenue‑generating product.

Restrictions & requirements

  • Attribution: The README credits stevenbucaille and the original paper. Include a citation to the SuperPoint paper when publishing results.
  • No warranty: The model is provided “as‑is”, without any guarantee of correctness or suitability for a particular purpose.
  • Modification: You may fine‑tune the model for internal use, but redistribution of the altered weights should respect the original “other” licence.

Pre-loaded AI models. Ready to run.

Skip the downloads. Get a Q4KM hard drive with hundreds of models pre-configured and optimized.

Shop Q4KM Hard Drives