(img1, img2, window_size=11, size_average=True)
| 381 | |
| 382 | |
| 383 | def ssim(img1, img2, window_size=11, size_average=True): |
| 384 | (_, channel, _, _) = img1.size() |
| 385 | global window |
| 386 | if window is None: |
| 387 | window = create_window(window_size, channel) |
| 388 | if img1.is_cuda: |
| 389 | window = window.cuda(img1.get_device()) |
| 390 | window = window.type_as(img1) |
| 391 | return _ssim(img1, img2, window, window_size, channel, size_average) |
no test coverage detected