/learned-frame-prediction

Predicts next video frame using 8 grayscale previous frames

Primary LanguagePython

LEARNED FRAME PREDICTION

Predicts next video frame using 8 grayscale previous frames. Requires Pytorch.

Model file and test videos can be downloaded from here: https://drive.google.com/open?id=1vhV9OYC_T8ED2BBTzLXAiaJw8aP2SxQm

Extract "Videos.zip" into project folder. Copy model file "MSE_only.pt" into "Models" directory.

Simply use the following line to run the code:

python run.py