Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

为什么训练得到的指标PSNR、SSIM=0 #5

Open
Young-Zzz opened this issue Jun 23, 2022 · 2 comments
Open

为什么训练得到的指标PSNR、SSIM=0 #5

Young-Zzz opened this issue Jun 23, 2022 · 2 comments

Comments

@Young-Zzz
Copy link

田教授您好,请问如何训练ACNet呢,下载readme分享的DIV2K数据集后,div2h.py生成DIV2.h5文件,填入train的--train_data_path,设定其他参数后,开始训练,但是最终的结果为PSNR、SSIM均等于0

@hellloxiaotian
Copy link
Owner

hellloxiaotian commented Jun 23, 2022 via email

@Young-Zzz
Copy link
Author

田教授您好,请问train用到的solver.py文件的第100行,

psnr = self.evaluate("dataset/Urban100", scale=cfg.scale, num_step=self.step)

这里调用了self.evaluate()函数,

def evaluate(self, test_data_dir, scale=2, num_step=0):def evaluate(self, test_data_dir, scale=2, num_step=0):

请问这里是不是需要指定cfg.test_data_dir,但是在train.py中没有这个参数输出要求,是我理解错了吗?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants