YangTang0072的博客

原创 Restormer复现

输入：“export PYTHONPATH=$PYTHONPATH:~/Restormer/”后再使用训练代码重新开始训练。在终端中运行python test.py，获取result文件。cd到Deraining，修改test.py里的参数。这里直接使用的云服务器，根据作者介绍完成1-4步。作者提供MATLAB方法，这里给出py方法。不采用渐进式训练，并且batch设置为1。三、测试及获取PSNR、SSIM。根据任务选择不同的训练代码。输入作者提供的代码开始训练。2、获取PSNR、SSIM。