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/> <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 ## Citation
If you find this code useful for your research, please use the following BibTeX entry. If you find this code useful for your research, please use the following BibTeX entry.
``` ```bibtex
@article{sun2021loftr, @article{sun2021loftr,
title={{LoFTR}: Detector-Free Local Feature Matching with Transformers}, title={{LoFTR}: Detector-Free Local Feature Matching with Transformers},
author={Sun, Jiaming and Shen, Zehong and Wang, Yuang and Bao, Hujun and Zhou, Xiaowei}, author={Sun, Jiaming and Shen, Zehong and Wang, Yuang and Bao, Hujun and Zhou, Xiaowei},

Loading…
Cancel
Save