ADD image pair info on ScanNet.

chiebot
JiamingSuen 4 years ago
parent 5e5c6fcc06
commit a82c40a042
  1. 36
      README.md

@ -142,12 +142,46 @@ For visualizing the results, please refer to `notebooks/visualize_dump_results.i
<br/>
### Image pair info for training on ScanNet
You can download the data at [here](https://drive.google.com/file/d/1fC2BezUSsSQy7_H65A0ZfrYK0RB3TXXj/view?usp=sharing).
<details>
<summary>[data format]</summary>
```python
In [14]: npz_path = './cfg_1513_-1_0.2_0.8_0.15/scene_data/train/scene0000_01.npz'
In [15]: data = np.load(npz_path)
In [16]: data['name']
Out[16]:
array([[ 0, 1, 276, 567],
[ 0, 1, 1147, 1170],
[ 0, 1, 541, 5757],
...,
[ 0, 1, 5366, 5393],
[ 0, 1, 2607, 5278],
[ 0, 1, 736, 5844]], dtype=uint16)
In [17]: data['score']
Out[17]: array([0.2903, 0.7715, 0.5986, ..., 0.7227, 0.5527, 0.4148], dtype=float16)
In [18]: len(data['name'])
Out[18]: 1684276
In [19]: len(data['score'])
Out[19]: 1684276
```
`data['name']` is the image pair info, organized as [`scene_id`, `seq_id`, `image0_id`, `image1_id`].
`data['score']` is the overlapping score defined in [SuperGlue](https://arxiv.org/pdf/1911.11763) (Page 12).
</details>
## Citation
If you find this code useful for your research, please use the following BibTeX entry.
```
```bibtex
@article{sun2021loftr,
title={{LoFTR}: Detector-Free Local Feature Matching with Transformers},
author={Sun, Jiaming and Shen, Zehong and Wang, Yuang and Bao, Hujun and Zhou, Xiaowei},

Loading…
Cancel
Save