zhang152267/dtcn

dtcn's Issues

执行代码时的报错

老师，您好！
非常抱歉在您百忙之中打扰您。首先，非常感谢您发表了优秀文章并提供了公开的代码，对我有很大的启发和帮助。可是，我在复现您的论文时遇到了以下问题：
1，在运行train.py文件时，第187行报错：UnboundLocalError: local variable 'input_train' referenced before assignment，即# log the images部分。我的理解是：## epoch training start 部分中定义的input_train为局部变量，导致# log the images部分无法调用。
2，我尝试了将# log the images部分代码缩进一个tab，便没有报错，我不知道这样是否正确。但是，这样去噪结果并不理想，PSNR和SSIM仅为11.73和0.344（epoch为100），而增加epoch后效果并没有明显改善（分别测试了epoch等于500和60000）。
希望您能给出宝贵的建议，期待您的回复，谢谢！
祝 身体健康！