imxieyi / waifu2x-ios-model-converter

Apache License 2.0
56 stars 3 forks source link

Model support request:RealSR、ATD-SR #15

Open Miracle-Sakuno opened 2 weeks ago

Miracle-Sakuno commented 2 weeks ago

RealSR DF2K

RealSR DPED

RealSR DF2K (JPEG)

These three models are said to be ESRGAN architecture.

May I ask if the author plans to use the RealSR model as a built-in support model? This way there are more parameters that can be adjusted.I saw that some App projects for image super-resolution also use RealSR as a built-in model. I tested it and the effect was pretty good. RealSR RealSR-NCNN-Android

ATD_SRx4

There is also a model of ATD architecture. I wonder if this can be supported?

The licenses of the above models are Apache2.0.

imxieyi commented 3 days ago

The 3 RealSR models have been added. And pre-converted models are available in Google Drive.

What kind of parameters of RealSR models would you like to adjust?

ATD support is not feasible due to Core ML issue:

ValueError: Core ML only supports tensors with rank <= 5. Layer "x.7", with type "reshape", outputs a rank 6 tensor.
Miracle-Sakuno commented 3 days ago

What I'm wondering is whether the RealSR model can adjust the magnification and noise reduction strength.

Also, thank you very much for your work.