tensor-mutator / StimuliNet

A FlowNet2.0 library

A FlowNet2.0 implementation with TensorFlow

Path to MPI Sintel dataset

flownet2/flow-dataset/flow_dataset/mpi_sintel/MPISintel/

Installation steps

(flownet2) poetry install

Intitiating training of FlowNetC with performance visualization

(flownet2) poetry run python3.6 -m flownet2.FlowNetC --train "(64, 64)"
(flownet2) tensorboard --logdir flownet2/FlowNetC/weights

Intitiating training of FlowNetCS with performance visualization

(flownet2) poetry run python3.6 -m flownet2.FlowNetCS --train "(64, 64)"
(flownet2) tensorboard --logdir flownet2/FlowNetCS/weights

Intitiating training of FlowNetCSS with performance visualization

(flownet2) poetry run python3.6 -m flownet2.FlowNetCSS --train "(64, 64)"
(flownet2) tensorboard --logdir flownet2/FlowNetCSS/weights

Intitiating training of FlowNetSD with performance visualization

(flownet2) poetry run python3.6 -m flownet2.FlowNetSD --train "(64, 64)"
(flownet2) tensorboard --logdir flownet2/FlowNetSD/weights

Intitiating training of FlowNet2.0 with performance visualization

(flownet2) poetry run python3.6 -m flownet2 --train "(64, 64)"
(flownet2) tensorboard --logdir flownet2/weights