Python scripts for performing 2D feature detection and tracking using the KP2D model in Pytorch
Python scripts for performing 2D feature detection and tracking using the KP2D model in Pytorch
Original images:https://github.com/liruoteng/OpticalFlowToolkit/blob/master/data/example/KITTI/
pip install -r requirements.txt
pip install pafy youtube_dl=>2021.12.17
Download the original models KeypointNet and KeypointResnet and extract them into the models folder.
The original repository contains additional code to train the models in Pytorch. This repository uses part of that code to make it easier to use the model in videos, images and webcamera.
python image_feature_matching.py
python image_feature_detection_conf.py
python video_feature_tracking.py
python webcam_feature_tracking.py
Original video: https://youtu.be/zP-gTCp5Kac