spmallick / learnopencv

Learn OpenCV : C++ and Python Examples
https://www.learnopencv.com/
20.67k stars 11.52k forks source link

Run RAFT in a Docker container #865

Closed SarvagyaVaish closed 7 months ago

SarvagyaVaish commented 7 months ago

I had a hard time setting up my local machine to run RAFT. To make it easier for the next person, I created a Docker image that runs a slightly modified version of the inference.py file on two provided frames. It already has Pytorch and all the other necessary dependecies installed.

Open to making tweaks to it if necessary. Please let me know :)

vikasgupta-github commented 7 months ago

Thanks @SarvagyaVaish