search

Zheng222 / IMDN

Lightweight Image Super-Resolution with Information Multi-distillation Network (ACM MM 2019, Winner Award of ICCVW AIM 2019 Constrained SR Track1&Track2)
418 stars 69 forks source link

About parameters #8

Closed SanSLee closed 4 years ago

SanSLee commented 4 years ago

Hello! I want to ask how you caculate the parameters of IDN? I use torchsummaryX to caculate the model size, but the parameters is about 677k.

Zheng222 commented 4 years ago

@UESTCshiny Hello, you can calculate the parameters of IDN manually through Eq. 9 in IMDN. Please note that the input channel of IDN is 1, and the output channel is also 1. I recommend that you use thop in PyTorch to calculate model parameters and FLOPs.