(self, window_size=11, size_average=True)
| 353 | |
| 354 | class SSIM(torch.nn.Module): |
| 355 | def __init__(self, window_size=11, size_average=True): |
| 356 | super(SSIM, self).__init__() |
| 357 | self.window_size = window_size |
| 358 | self.size_average = size_average |
| 359 | self.channel = 1 |
| 360 | self.window = create_window(window_size, self.channel) |
| 361 | |
| 362 | def forward(self, img1, img2): |
| 363 | (_, channel, _, _) = img1.size() |
nothing calls this directly
no test coverage detected