Improvement: Override middle blocks , Improvement:Allow block types o…

TrystAI / restorers

Restorers provide out-of-the-box TensorFlow implementations of SoTA image and video restoration models for tasks such as low-light enhancement, denoising, deblurring, super-resolution, etc.

https://wandb.me/low-light

36 stars 0 forks source link

Improvement: Override middle blocks , Improvement:Allow block types o… #57

Closed Samarendra109 closed 1 year ago

Samarendra109 commented 1 year ago

Fixes #50 #51

Changes proposed by this PR :-

added a new method create_middle_blocks
added a new method get_block
added a new parameter to NAFNet constructor
added new tests for the different block types in NAFNet

Request for Review: @soumik12345, @SauravMaheshkar

Samarendra109 commented 1 year ago

@SauravMaheshkar , @soumik12345 please have a look. The changes are very simple and should be quick enough to review. Thanks.