Skip to content

Latest commit

 

History

History
81 lines (53 loc) · 2.2 KB

README.md

File metadata and controls

81 lines (53 loc) · 2.2 KB

Build

📝 torch_cpp


🎛 Dependencies


sudo apt-get install -y --no-install-recommends \
    libopencv-dev

🔨 How to Build


# build library
make default

# build examples
make apps -j`nproc`

# build gpu examples
make gpu_apps -j`nproc`

🏃 How to Run


Usage
  • Script pre-trained superpoint and superglue weights
git submodule update --init --recursive
export ROOT_DIR="$(pwd)"
cd $ROOT_DIR/scripts/superglue/SuperGluePretrainedNetwork/
git apply ../jit_patch.patch
python3 -m pip install -r $ROOT_DIR/scripts/superglue/SuperGluePretrainedNetwork/requirements.txt
cd $ROOT_DIR
python3 $ROOT_DIR/scripts/superglue/jit_superglue_model.py
python3 $ROOT_DIR/scripts/superglue/jit_superpoint_model.py
  • Test inference apps
./build/examples/match_images_superglue/match_images_superglue_app path/to/superpoint_model.pt path/to/superglue_model.pt ./data/images/VisionCS_0a.png ./data/images/VisionCS_0b.png

(back to top)

💎 References


(back to top)