spmallick / learnopencv

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

Run RAFT in a Docker container #865

Closed SarvagyaVaish closed 1 year ago

SarvagyaVaish commented 1 year 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 1 year ago

Thanks @SarvagyaVaish