├── LICENSE ├── README.md ├── compute_desc_eval.py ├── compute_repeatability.py ├── config ├── detection_repeatability.yaml ├── export_descriptors.yaml ├── homo_export_labels.yaml ├── magic_point_coco_train.yaml ├── magic_point_syn_train.yaml └── superpoint_train.yaml ├── dataset ├── coco.py ├── patch.py ├── synthetic_shapes.py └── utils │ ├── augmentation_legacy.py │ ├── homographic_augmentation.py │ ├── photometric_augmentation.py │ └── synthetic_dataset.py ├── export_descriptors.py ├── export_detections_repeatability.py ├── homo_export_labels.py ├── model ├── magic_point.py ├── modules │ └── cnn │ │ ├── cnn_heads.py │ │ └── vgg_backbone.py ├── superpoint.py └── superpoint_bn.py ├── requirements.txt ├── solver ├── descriptor_evaluation.py ├── detector_evaluation.py ├── loss.py └── nms.py ├── superpoint_bn.pth ├── superpoint_v1.pth ├── train.py ├── utils ├── archive.py ├── keypoint_op.py ├── log.py ├── params.py ├── plt.py ├── tensor_op.py └── warp_perspective.py └── visual_syn_pred.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /compute_desc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/compute_desc_eval.py -------------------------------------------------------------------------------- /compute_repeatability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/compute_repeatability.py -------------------------------------------------------------------------------- /config/detection_repeatability.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/detection_repeatability.yaml -------------------------------------------------------------------------------- /config/export_descriptors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/export_descriptors.yaml -------------------------------------------------------------------------------- /config/homo_export_labels.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/homo_export_labels.yaml -------------------------------------------------------------------------------- /config/magic_point_coco_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/magic_point_coco_train.yaml -------------------------------------------------------------------------------- /config/magic_point_syn_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/magic_point_syn_train.yaml -------------------------------------------------------------------------------- /config/superpoint_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/config/superpoint_train.yaml -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/patch.py -------------------------------------------------------------------------------- /dataset/synthetic_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/synthetic_shapes.py -------------------------------------------------------------------------------- /dataset/utils/augmentation_legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/utils/augmentation_legacy.py -------------------------------------------------------------------------------- /dataset/utils/homographic_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/utils/homographic_augmentation.py -------------------------------------------------------------------------------- /dataset/utils/photometric_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/utils/photometric_augmentation.py -------------------------------------------------------------------------------- /dataset/utils/synthetic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/dataset/utils/synthetic_dataset.py -------------------------------------------------------------------------------- /export_descriptors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/export_descriptors.py -------------------------------------------------------------------------------- /export_detections_repeatability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/export_detections_repeatability.py -------------------------------------------------------------------------------- /homo_export_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/homo_export_labels.py -------------------------------------------------------------------------------- /model/magic_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/model/magic_point.py -------------------------------------------------------------------------------- /model/modules/cnn/cnn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/model/modules/cnn/cnn_heads.py -------------------------------------------------------------------------------- /model/modules/cnn/vgg_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/model/modules/cnn/vgg_backbone.py -------------------------------------------------------------------------------- /model/superpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/model/superpoint.py -------------------------------------------------------------------------------- /model/superpoint_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/model/superpoint_bn.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver/descriptor_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/solver/descriptor_evaluation.py -------------------------------------------------------------------------------- /solver/detector_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/solver/detector_evaluation.py -------------------------------------------------------------------------------- /solver/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/solver/loss.py -------------------------------------------------------------------------------- /solver/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/solver/nms.py -------------------------------------------------------------------------------- /superpoint_bn.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/superpoint_bn.pth -------------------------------------------------------------------------------- /superpoint_v1.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/superpoint_v1.pth -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/train.py -------------------------------------------------------------------------------- /utils/archive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/archive.py -------------------------------------------------------------------------------- /utils/keypoint_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/keypoint_op.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/params.py -------------------------------------------------------------------------------- /utils/plt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/plt.py -------------------------------------------------------------------------------- /utils/tensor_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/tensor_op.py -------------------------------------------------------------------------------- /utils/warp_perspective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/utils/warp_perspective.py -------------------------------------------------------------------------------- /visual_syn_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shaofengzeng/SuperPoint-Pytorch/HEAD/visual_syn_pred.py --------------------------------------------------------------------------------