Autoregression.Pytorch

AR(2) auto regression model.

Support:

• AR(2)
• FFT decompose

Installation

git clone https://github.com/Fangyh09/Autoregression.Pytorch.git

Usage

python main.py

Screenshot-NoFFT local_autoregression

input img0, img1, img2

output img3

Screenshot- FFT local_autoregression

input img0, img1, img2

output img3

Screenshot-FFT

python main_image_fft.py

input img0, img1, img2

output img3

Screenshot-NoFFT

python main_image.py

input img0, img1, img2

output img3

TODOs

• Add fft decompose
• Add local conv
• Add optical flow

History

Credits

Some codes are from @pysteps. Thanks @ssim-pytorch, @pysteps