ailia-models
ailia-models copied to clipboard
ADD AnimeInterp
https://github.com/lisiyao21/AnimeInterp Our code is released under MIT License.
cuda extensionを使用しているが、使わないモデルも提供されているので、そちらを使用するようにする。
〇 configs/config_test_w_sgm.py
model = 'AnimeInterp'
↓
model = 'AnimeInterpNoCupy'
最近のバージョンではscikit-imageのメソッド名が変わっているため、修正する。
〇 test_anime_sequence_one_by_one.py
from skimage.measure import compare_psnr, compare_ssim
↓
from skimage.metrics import structural_similarity as compare_ssim
from skimage.metrics import peak_signal_noise_ratio as compare_psnr
grid_sampleを使っており、現時点でエクスポート不可
RuntimeError: Exporting the operator grid_sampler to ONNX opset version 11 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub.
サンプル画像は Japan_v2_3_052198_s2