zhangbaijin / memorynet

Code for paper: Memory Augment is All Your Need for image restoration(cloud,rain,shadow removal, low-light image enhancement, image deblur)即插即用提点的记忆模块

Hi there 👋

• 🔭 I’m currently a Ph.D student in SJTU(IWIN Lab), now i am a intern research in DAMO ACADEMIC (达摩院).
• 🌱 I’m currently learning LLM/MLLM, explainable attention.
• 💬 Ask me about anything
• 📫 How to reach me: [email protected]. 微信：SemiZxf
• 🌱 homepage:(zhangbaijin.github.io)
• 💬 Google sclolar: Google scholar

memorynet's Issues

您好，预训练去云模型网盘地址失效，可否提供新的地址，谢谢！

您好，预训练去云模型网盘地址失效，可否提供新的地址，谢谢！

找不到文件

运行后报错，找不到文件
PS C:\Users\dell\Desktop\yhdd\test_cloud\MemoryNet-main> python demo.py --task Deraining --input_dir ./Test-image-RICE/cloudy_image --result_dir ./result/1.png
Traceback (most recent call last):
File "C:\Users\dell\Desktop\yhdd\test_cloud\MemoryNet-main\demo.py", line 52, in
load_file = run_path(os.path.join(task, "MPRNet.py"))
File "D:\yhdd\language\python\lib\runpy.py", line 288, in run_path
code, fname = _get_code_from_file(run_name, path_name)
File "D:\yhdd\language\python\lib\runpy.py", line 252, in _get_code_from_file
with io.open_code(decoded_path) as f:
FileNotFoundError: [Errno 2] No such file or directory: 'C:\Users\dell\Desktop\yhdd\test_cloud\MemoryNet-main\Deraining\MPRNet.py'

demo的model是？

你好！我想直接用您train好的model来试试，但遇到了一些问题，还请指教！

To test the pre-trained models of Decloud on your own images, run

python demo.py --task Task_Name --input_dir path_to_images --result_dir save_images_here

这里必须指定的task，是Deblurring, Denoising和Deraining任选一个吗？它们对应的model是您提供的哪一个呢？我download下来的有三个model_best.pth, model_epoch_250_250.pth, model_latest.pth，不知该如何和这些task对应。谢谢！

关于训练的问题

您好，运行环境和原作者的运行环境是一样的吗？
在训练的时候是否出现了下面这种问题：

@

FileNotFoundError: [Errno 2] No such file or directory: 'C:\\Users\\Waqas\\Downloads\\Constractive-learning-shadow-removal-main\\Constractive-learning-shadow-removal-main\\Deblurring\\ MPRNet.py'

FileNotFoundError: [Errno 2] No such file or directory: 'C:\Users\Waqas\Downloads\Constractive-learning-shadow-removal-main\Constractive-learning-shadow-removal-main\Deblurring\
MPRNet.py'

In MemoryNet.py, MemoryNet() forward(). How to get the x2 and x3, thif function input is x3_img

请问对应的论文是什么

好像没搜到本代码对应的论文