Donkeycar neural network model visualization built on top of https://github.com/raghakot/keras-vis
- OpenCV
- Conda
- Trained neural network
- Donkeycar records tub
- Install opencv (OS Specific)
- Install conda environment
conda env create -f environment.yml
- Activate conda env
conda activate donkey-vis
- For now:
pip install git+https://github.com/raghakot/keras-vis.git --upgrade
(as long as some bug has been fixed) - Check that the installed tensorflow version matches with the version model was trained on
conda list | grep tensorflow
(Defaults to 1.11.0) - If it doesn't match, update tensorflow version
conda install tensorflow-gpu==1.11.0
- Run visualization with
python visualize --model ./model --tub ./records
(python visualize -h for more parameters
) - Kill with ctrl-c when enough frames exported