损失函数:损失函数用多任务学习,针对三类失真图像同时训练三种图像重建任务,如下式,三项分别对应三类重建任务的损失。 损失函数不仅考虑像素级损失(MSE),还要考虑结构性损失(SSIM)和梯度信息,如下式。 三项分别为MSE用于计算像素级的损失、SSIM用于计算结构损失、TV用于计算总变分损失减少图像的虚假边缘和噪声等失真,同时...
PATH] [--score-path PATH] DIR DIR PyTorch EBSNetMEFNet positional arguments: DIR path to testset DIR path to models optional arguments: -h, --help show this help message and exit --data-type 'night' or 'day' --results path to save results --score-path path to save psnr and ssim...