Skip to content

Commit

Permalink
Add OpenCV as dependency, add reference to install compatible version…
Browse files Browse the repository at this point in the history
…s of pytorch with CUDA (cvg#29)

* Add OpenCV as dependency and pytorch-CUDA compatible reference link to README

* update readme.

* update readme.

* update readme.

* update tp-lsd installation in readme.

* minor update.

* fix tp-lsd link.

Co-authored-by: B1ueber2y <[email protected]>
  • Loading branch information
MarkYu98 and B1ueber2y authored Nov 29, 2022
1 parent 28b87cf commit c52667f
Showing 1 changed file with 16 additions and 13 deletions.
29 changes: 16 additions & 13 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,20 @@

## Dependencies
* CMake >= 3.17
* Python 3.9
* COLMAP
* COLMAP [[Guide](https://colmap.github.io/install.html)]
* HDF5
```bash
sudo apt-get install libhdf5-dev
```
* OpenCV (only for installing [pytlbd](https://github.com/B1ueber2y/limap-internal/blob/main/requirements.txt#L33))
```bash
sudo apt-get install libopencv-dev libopencv-contrib-dev libarpack++2-dev libarpack2-dev libsuperlu-dev
```

* Python 3.9 + required packages
```bash
git submodule update --init --recursive
sudo apt-get install libhdf5-dev
python -m pip install torch==1.11.0 torchvision==0.12.0
python -m pip install torch==1.11.0 torchvision==0.12.0 # Refer to https://pytorch.org/get-started/previous-versions/ to install packages that are compatible with your CUDA
python -m pip install -r requirements.txt
```

Expand All @@ -21,7 +28,7 @@ python setup.py develop

or you can also use ``pip`` alternatively:
```
python -m pip install -ve .
python -m pip install -ve .
```

## Quickstart
Expand Down Expand Up @@ -55,17 +62,13 @@ The following line detectors are currently supported:
- [LSD](https://github.com/iago-suarez/pytlsd)
- [SOLD2](https://github.com/cvg/SOLD2)
- [HAWPv3](https://github.com/cherubicXN/hawp)

The line detector [TP-LSD](https://github.com/Siyuada7/TP-LSD) can be additionally used, but needs a separate installation. You will need to switch to GCC 7 for the following compilation (but can use again GCC 9 at test time):
```bash
python -m pip install -e ./third-party/TP-LSD/tp_lsd/modeling/DCNv2
python -m pip install -e ./third-party/TP-LSD
```
- [TP-LSD](https://github.com/Siyuada7/TP-LSD)

The following line descriptors/matchers are currently supported:
- [LBD](https://github.com/iago-suarez/pytlbd)
- [SOLD2](https://github.com/cvg/SOLD2)
- [LineTR](https://github.com/yosungho/LineTR)
- [L2D2](https://github.com/hichem-abdellali/L2D2)
- Nearest neighbor matching of the endpoints with [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork)
- Matching of the endpoints with [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork) + [SuperGlue](https://github.com/magicleap/SuperGluePretrainedNetwork)
- Endpoint matching with [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork) + Nearest Neighbors
- Endpoint matching with [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork) + [SuperGlue](https://github.com/magicleap/SuperGluePretrainedNetwork)

0 comments on commit c52667f

Please sign in to comment.