Skip to content

Latest commit

 

History

13 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FeatureTracker

What this repository for

Detect, match and track different type features

Current type of feture points

  • ORB feature points
  • SuperPoint feature points

Usage

PreInstall

install libpytorch: Install from source with Anaconda Set libpytorch cmake path at CMakeLists.txt

Dependence Info

pytorch: 1.1.0

cuda/cudatoolkit: 9.0

Build

$ cd FeatureTracker
$ mkdir build
$ cd build && cmake ..
$ make -j4

Run ORB feature points

$ ./FeatureTracker --image_dir /path/to/images_dir --type orb [--image_suffix .jpg]

SuperPoint feature points

Download superpoint model
$ cd utils
$ wget https://github.com/MagicLeapResearch/SuperPointPretrainedNetwork/raw/master/superpoint_v1.pth
$ python pytorch_model_convert.py superpoint_v1.pth
$ cd ../build
$ ./FeatureTracker --image_dir /path/to/image_dir --model_path ../utils/super_point_scripted_model.pt --type sp [--image_suffix .jpg --H 480 --W 640]

Kitti:

ORB Tracker

SuperPoint Tracker

Resized image to 414, 125 (width/3, height/3)

Original size image

About

Features detection, match and track module. Supported features: orb and superpoints

Resources

Stars

30 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages