Skip to content

Learning to estimate optical flow with FlowNet using TensorFlow

Notifications You must be signed in to change notification settings

SeokjuLee/TF_FlowNet

Repository files navigation

TF_FlowNet

Model links

Dependencies

conda create -n SK_Week4_sfm tensorflow-gpu=1.8 ipython jupyter pillow ipykernel  gdbm anaconda -y

source  activate SK_Week4_sfm

conda install -c conda-forge matplotlib -y
conda install -c menpo opencv
conda install -c anaconda scipy
conda install tensorflow-gpu=1.8

python -m ipykernel install  --user --name SK_Week4_sfm --display-name "[SK_Week4_sfm]"
jupyter notebook

Usage

  1. Activate conda env
cd ~/sk_week4
source activate SK_Week4_RCV
  1. Clone or pull git
git clone https://github.com/SeokjuLee/TF_FlowNet
cd TF_FlowNet

cd TF_FlowNet
git reset --hard
git pull
  1. Download dataset
cd dataset/FlyingChairs/data
chmod +x download_dataset.sh
./download_dataset.sh
cd ../../..
  1. Download models
cd model/flownet_simple/755
chmod +x download_model.sh
./download_model.sh
cd ../../..

cd model/flownet_simple/333
chmod +x download_model.sh
./download_model.sh
cd ../../..
  1. Run test_flownet_simple.py
CUDA_VISIBLE_DEVICES=0 python3 test_flownet_simple.py

About

Learning to estimate optical flow with FlowNet using TensorFlow

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages