Source code for spectral RNN learning with optimizable window functions.
The window functions are available in src/window_learning.py
.
The code has been tested using Tensorflow 1.10.0 on Ubuntu-Linux.
-
To recreate the synthetic experiments run
synthetic_signal_test.py
after adjusting the hyperparameters as described in the paper. The results may be plotted by usingsynthetic_signal_plot.py
after adjusting the log file path in that file. -
To run the power prediction experiments, download: http://www.wolter.tech/wordpress/wp-content/uploads/2019/02/power_data.zip And extract it in the
src/power_experiments
folder. Then runpower_train_exp.py
from within the same directory with the desired parameters. -
The mocap experiments use the human3.6m data set available at http://vision.imar.ro/human3.6m/ . We use the D3 Position files from the "by subject category". After downloading and pickling the data run
mocap_train_exp.py
to repeat the experiments in the paper.
A preprint https://arxiv.org/pdf/1812.05645.pdf, and the springer version https://link.springer.com/chapter/10.1007/978-3-030-61609-0_65 are available. If you find this work useful, please consider citing the paper:
@inproceedings{wolter2020spectral,
title={Sequence Prediction using Spectral RNNs},
author={Wolter, Moritz and Gall, Juergen and Yao, Angela},
booktitle={29th International Conference on Artificial Neural Networks},
year={2020}
}
In the demo, the desired behavior is shown on the left, while the right side depicts the network predictions. The red and blue colored stick figures are context; the green and yellow figures show the ground truth and network output. The selection is not balanced. I have chosen examples which worked well.
This work was funded by the Deutsche Forschungsgemeinschaft (DFG, German Research Foundation) YA 447/2-1 and GA 1927/4-1 (FOR2535 Anticipating Human Behavior) as well as by the National Research Foundation of Singapore under its NRF Fellowship Programme [NRF-NRFFAI1-2019-0001].