No module named 'models.basicblock'

cszn / USRNet

Deep Unfolding Network for Image Super-Resolution (CVPR, 2020) (PyTorch)

https://cszn.github.io/

865 stars 115 forks source link

No module named 'models.basicblock' #6

Closed taaa22 closed 4 years ago

taaa22 commented 4 years ago

Hi, I'm getting some error

File "main_test_realapplication.py", line 19, in from models.network_usrnet import USRNet as net File "/content/USRNet/models/network_usrnet.py", line 3, in import models.basicblock as B ModuleNotFoundError: No module named 'models.basicblock'

I think file basicblock.py is missing which should be located at models folder

BlueAmulet commented 4 years ago

Grab it from here https://github.com/cszn/DPSR/blob/master/models/basicblock.py

taaa22 commented 4 years ago

I already did that, thanks for sharing.

I'm wondering if the model is denoising the image or not because I tried it, it works fine on SR but it didn't denoise the image even when I increase the noise level.

cszn commented 4 years ago

Added.