ailia-models icon indicating copy to clipboard operation
ailia-models copied to clipboard

ADD AnimeInterp

Open kyakuno opened this issue 3 years ago • 4 comments

https://github.com/lisiyao21/AnimeInterp Our code is released under MIT License.

kyakuno avatar Nov 22 '21 07:11 kyakuno

cuda extensionを使用しているが、使わないモデルも提供されているので、そちらを使用するようにする。

〇 configs/config_test_w_sgm.py

model = 'AnimeInterp'
↓
model = 'AnimeInterpNoCupy'

ooe1123 avatar Dec 03 '21 13:12 ooe1123

最近のバージョンでは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

ooe1123 avatar Dec 03 '21 13:12 ooe1123

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.

ooe1123 avatar Dec 03 '21 14:12 ooe1123

サンプル画像は Japan_v2_3_052198_s2

ooe1123 avatar Dec 03 '21 16:12 ooe1123