chaiNNer-org / spandrel

Spandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting. Based on the model support implemented in chaiNNer.
MIT License
103 stars 7 forks source link

Retinexformer support? #269

Closed zelenooki87 closed 1 month ago

zelenooki87 commented 1 month ago

https://github.com/caiyuanhao1998/Retinexformer Really nice project. Thank you very much in advance!

zelenooki87 commented 1 month ago

Hey, the FiveK model does some really beautiful tone and color correction, it's great! I converted it to an ONNX model from Retinexformer, but I'm running into some errors in Chainner when I try to load it with cropped images that have non-standard dimensions. It seems to work fine on everything else, though. It would be fantastic if you could incorporate this architecture, especially since it doesn't look like it needs any extra dependencies. Thanks again for everything! Best, Milan

zelenooki87 commented 1 month ago

Could you please answer, is it doable to add?

Thanks

joeyballentine commented 1 month ago

Anything is possible, it just matters if we're willing to do the work to add it. It takes time to properly add support for an arch

zelenooki87 commented 1 month ago

Okay, I will patiently wait and hope then. :)

zelenooki87 commented 1 month ago

Anything is possible, it just matters if we're willing to do the work to add it. It takes time to properly add support for an arch

Can I get a contact email? I'm considering a donation for the implementation of Retinexformer. Thank you.