CatHand Blog

アプリ開発やMac弄り

TecoGANで動画の超解像度化

環境:

Anaconda Individual Edition をインストール。

www.anaconda.com

環境設定とTecoGANのインストール等。

% conda create -n tecog python=3.7.7 numpy scipy scikit-image matplotlib pandas Keras torchvision ipython tensorflow=1.13.1
% conda activate tecog
% git clone https://github.com/thunil/TecoGAN.git
% cd TecoGAN
% pip3 install -r requirements.txt
% python runGan.py 0

これで

% python runGan.py 1

とすると、 TecoGAN/LR/calendar/ 内の画像が TecoGAN/results/calendar/超解像度化されて保存されます。

pythonよくわかんないので、これをそのまま利用することにします。

  • 動画を連番画像にして TecoGAN/LR/calendar/ へ書き出し
  • python runGan.py 1
  • TecoGAN/results/calendar/内の超解像度化された画像をffmpegで動画にする

こんなスクリプトになりました。

gist.github.com

  • fpsは維持するようにしています。
  • スクリプト内でanacondaの環境設定をしないと動かなかったので入れています(ようわからん)
./upscale input.mp4 output.mp4

というかんじで変換できます。