NameError: name 'AG_Net' is not defined

[tsorewilly]

Hi,

I am trying to understand your code after reading the paper. There are several issues I am trying to deal with. First, running the code directly from main.py, I got an error stating "NameError: name 'AG_Net' is not defined". Also, running the code directly from models.py gave another error stating that "ModuleNotFoundError: No module named 'guided_filter_pytorch.guided_filter_attention'". Thus, I am not sure which one is the base file. Also, I tried to search google for more guidance but I didn't find any.

Could you please let me know what to fix to make it work?