NAFNet icon indicating copy to clipboard operation
NAFNet copied to clipboard

enc_blk_nums middle_blk_num dec_blk_nums

Open Ken1256 opened this issue 2 years ago • 4 comments

Different tasks need required different settings?

Ken1256 avatar Jul 01 '22 13:07 Ken1256

Hi, Ken1256,

For different tasks, the differentiated design of the network structure is not our original creation, we set this up empirically. In essence, we believe that although denoise/deblur are both called "image restoration tasks", they require different capabilities of the network, such as: Work related to deblur usually emphasizes the "receptive field", while work related to denoise does not seem to emphasize this. I hope that alleviates your concerns.

Thanks.

mayorx avatar Jul 04 '22 03:07 mayorx

Why enc_blk_nums[3] "receptive field" better than middle_blk_num? How does the deblur + denoise task allocate 36 blocks?

Ken1256 avatar Jul 05 '22 14:07 Ken1256

These are the result of trade-offs between different quantities. E.g. the "receptive field", "# of parameters", "simplicity", and etc. "36 blocks" is also the result of a trade-off between performance and latency.

These results are not set in stone either, and we haven't fully explored the effects of these quantities, so this is an open question.

mayorx avatar Jul 05 '22 15:07 mayorx

Can enc_blk_nums: [1, 1, 1, 28] be understood as having four encoder modules, the first, second and third encoder modules contain 1 NAFBlock, and the fourth encoder module contains 28 NAFBlocks? What is the reason for this design?

Amy-214 avatar Sep 03 '23 03:09 Amy-214