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

Add support for RetinexFormer #270

Closed RunDevelopment closed 1 month ago

RunDevelopment commented 1 month ago

Closes #269.

This is our first low-light model, so that's quite interesting.

We also need to add more ways to tile images in the future. Low-light models (and similar) rely on global image information to remap colors, so chaiNner's normal tiling methods are pretty much useless. Right now, I used the same trick the RetinexFormer authors used for really large images: interlacing. This isn't a silver bullet though. More on that in the future.