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

Young-Zzz commented 2 years ago

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

hellloxiaotian commented 2 years ago

检查一下训练图像和验证图像读取路径和图像格式（.jpg,.png）。

祝好！春伟

在 2022-06-23 13:13:46，"User-Xzm" @.***> 写道：

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

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

Young-Zzz commented 2 years ago

田教授您好，请问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中没有这个参数输出要求，是我理解错了吗？

hellloxiaotian / ACNet

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