(self, window_size=11, size_average=True)
| 192 | |
| 193 | class SSIMLoss(torch.nn.Module): |
| 194 | def __init__(self, window_size=11, size_average=True): |
| 195 | super(SSIMLoss, self).__init__() |
| 196 | self.window_size = window_size |
| 197 | self.size_average = size_average |
| 198 | self.channel = 1 |
| 199 | self.window = create_window(window_size, self.channel) |
| 200 | |
| 201 | def forward(self, img1, img2): |
| 202 | (_, channel, _, _) = img1.size() |
nothing calls this directly
no test coverage detected