swz30 / MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.
Other
667 stars 96 forks source link

Some questions about your code #34

Closed Spacei567 closed 2 years ago

Spacei567 commented 2 years ago

Hi, Thanks you sharing this work! The code in your MIRNet_model.py in SKFF, batch_size = inp_feats[0].shape[0]

I want to ask you that the inp_feats are [B, C, H, W], inp_feats[0] indicates that a batch's [C, H, W], so inp_feats[0].shape[0] indicates that C, but why your code it indicates batch_size?

Spacei567 commented 2 years ago

The question was solved.