ONNX
aidge
File size: 774 Bytes
2a7c734
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
---
license: mit
library_name: onnx
tags:
  - onnx
  - aidge
---
# SuperPoint ONNX

SuperPoint model exported to ONNX by PrunaAI. The base SuperPoint model is adapted from https://github.com/rpautrat/SuperPoint repository `superpoint_pytorch.py` script and also uses the corresponding weights from the repository.

## Model

- Source model: SuperPoint
- Export format: ONNX
- ONNX opset: 18
- Input: `inputs`, shape `[1, 1, 768, 1024]`
- Output: `keypoints`, shape `[1, 512, 2]`; `scores`, shape `[1, 512]`; `descriptors`, shape `[1, 512, 256]`

## Usage in Aidge

The model is intended to be used as the base model in the Aidge SuperPoint pruning tutorial.

## License

The original SuperPoint model is provided in https://github.com/rpautrat/SuperPoint under MIT license.