torch_book.metrics.utils#

Classes#

CropBorder

裁掉图像的边界

Module Contents#

class torch_book.metrics.utils.CropBorder(size: int = 0, *args, **kwargs)[源代码]#

Bases: torch.nn.Module

裁掉图像的边界

参数:

size -- 图像每条边缘裁剪的像素。这些裁剪掉的像素不参与 PSNR 的计算。默认值为 0。

size = 0#
forward(x)[源代码]#