(img1, img2, win_size=11, data_range=1, size_average=True)
| 148 | |
| 149 | |
| 150 | def ssim(img1, img2, win_size=11, data_range=1, size_average=True): |
| 151 | (_, channel, _, _) = img1.size() |
| 152 | window = create_window(win_size, channel) |
| 153 | |
| 154 | if img1.is_cuda: |
| 155 | window = window.cuda(img1.get_device()) |
| 156 | window = window.type_as(img1) |
| 157 | |
| 158 | return _ssim(img1, img2, window, win_size, channel, size_average) |
| 159 | |
| 160 | |
| 161 | def elementwise_charbonnier_loss( |
nothing calls this directly
no test coverage detected