A pair of subsequent frames is selected to obtain optical flow. This flow image is used to learn the velocity of the ego-car.
python 3.5.2 Tensorflow-Gpu==1.7 Keras==2.2.4
- tdg.py -> Has Training Data Generator Object for Neural Networks Train/Eval
- data_creator.py -> Has functions to preprocess videos and visualize results on video
- utils.py -> Side utility functions useful in multiple parts of the repository
- io_args.py -> Argument Parser for different inputs
- train_eval.py -> Performs training
- eval_pipeline.py -> Store model's inferred output on train/test data into pickle files
- rough_plots.py -> Visualize model results and store results as text files
- model_set.py -> Keras Model Definitions