No description, website, or topics provided.
Python Shell
Switch branches/tags
Nothing to show
Clone or download
Pull request Compare This branch is 5 commits ahead of quadjr:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
scripts
LICENSE
README.md
main.py
net.py

README.md

GRU-PredNet

This is a GRU version of PredNet based on quadjr/PredNet.

ConvLSTM is replaced by ConvGRU, the others are the same as the original version.

Note

Comparing output-images between this version and original one about images in image_02 in 2011_09_26_drive_0001_sync.zip of KITTI showed some differences, the effectiveness is under investigation.

Testd on

  • Ubuntu 14.04
  • Python 2.7.6
  • chainer 1.9.1
  • CUDA Toolkit 7.5
  • cuDNN v5

Usage

  1. Train
    On a CPU
    $ python main.py -i dataset/train_list.txt
    On a GPU 0 $ python main.py -i dataset/train_list.txt -g 0

Model files and some images will be generated in models/ and images/ directory.
Image suffix means x (input), y (predicted), z (correct).

  1. Test
    $ python main.py -i dataset/test_list.txt --test --initmodel models/???.model
    Plase specify the model file with option --initialmodel.
    Test result will be generated in images/ directory.
    Image suffix means x (input), y (predicted).