変なアーティファクトがあまり入らない超解像

RealESRGANも強力だけれど、平滑化が強すぎて細部が潰れてしまったり、他のモデルもクッキリはするけれど、アーティファクトが入ってしまったりします。

RDNは癖がない

なるべくシンプルに、しかししっかり超解像はする、そんな要望に比較的答えていると僕が個人的に思うモデルがRDNです。

↓超解像

使い方

インストール。

!git clone https://github.com/idealo/image-super-resolution
%cd image-super-resolution
!python setup.py install

推論。

from ISR.models import RDN
import numpy as np
from PIL import Image

img = Image.open('lr.jpg')
lr_img = np.array(img)
rdn = RDN(weights='psnr-small')
sr_img = rdn.predict(lr_img, by_patch_of_size=128)
result = Image.fromarray(sr_img)
result.save('result.jpg')

ちなみに、より強力なRDN largeモデル、ノイズ除去モデルと、RRDNGANも用意されています。

rdn_large = RRDN(weights='psnr-large')
rdn_noise = RRDN(weights='noise-cancel')
rrdn = RRDN(weights='gans')

🐣

フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com

機械学習、ARアプリ（Web/iOS）を作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium
GitHub