-
Notifications
You must be signed in to change notification settings - Fork 453
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
LiteFlowNet with cupy instead of FlowNet2 with compiled modules #63
base: master
Are you sure you want to change the base?
Conversation
Also replace mmcv progress bar with tqdm
no need to compile anything anymore
take changes from v1.1
replace flownet2 with liteflownet
reference original liteflow repository
Very thanks for you contribution and sorry for my late reply. I will consider to make a new branch for your request. Thanks again for your interest in our project. |
Thank you so much for this contribution! |
For anyone who comes across this thread in the future - here's how I was able to get the software to work (Manjaro Arch Linux 5.18 + CUDA 1.17 + Nvidia + without Anaconda). It's likely that you have to tweak some dependencies:
new_requirements.txt Download models and demos from + place them in the correct folder (e.g. models/pretrained_models - see original instruction)
Replace Slighty different call (because of network-default.pytorch and --LiteFlowNet)
|
I've replaced all the FlowNet2 files with corresponding LiteFlowNet implementations from the pytorch reimplementation. This implementation doesn't need the flownet2 modules but is based on CuPy. In return the compiling of the flownet2 modules is no longer a problem. Also LiteFlowNet is 30 times smaller in the model size and 1.36 times faster in the running speed and outperforms the FlowNet2 on the challenging Sintel final pass and KITTI benchmarks [1].
Quick note: I didn't mean to request merging into the master branch. I just wanted to let you know that when compiling the flownet2 modules fails there is another option. Maybe you can make a new LiteFlowNet branch or reference my fork in the FAQ's answer.
Here is the flamingo demo generated using the LiteFlowNet: