search

itberrios / CV_projects

General Computer Vision Projects
MIT License
40 stars 14 forks source link

RAFT dive notebooks appear to be broken #1

Open starsong98 opened 10 months ago

starsong98 commented 10 months ago

Hello,

Thank you for your amazing article https://towardsdatascience.com/optical-flow-with-raft-part-2-f0376a972c25 and releasing the code for it.

However, I have been unable to get the raft_dive.ipynb notebook to work. It appears that something is wrong with the module imports. Can you look into it? Or is there a specific order in which each notebook needs to be run?

Also, where can I find the test version of RAFT that you modified? From what I can tell, this repository seems to directly import the baseline version of RAFT and I cannot seem to find the modified test version of RAFT that you mentioned. Can you tell me where the motion feature is extracted from?

Thank you!

itberrios commented 10 months ago

Hello,

Thanks for pointing this out! I realized that the RAFT in this repo was a submodule to the original version. I have replaced with my modified version and now it should work. The RAFT_dive notebook is really an experimental thing I did to help improve my understanding of RAFT. It looks like I will need to refresh it at some point to make it easier to follow. Please try again and let me know if there are any issues, there's no specific order of which to follow the notebooks, but I would recommend this: