(window_size, channel)
| 47 | return gauss / gauss.sum() |
| 48 | |
| 49 | def create_window(window_size, channel): |
| 50 | _1D_window = gaussian(window_size, 1.5).unsqueeze(1) |
| 51 | _2D_window = _1D_window.mm(_1D_window.t()).float().unsqueeze(0).unsqueeze(0) |
| 52 | window = Variable(_2D_window.expand(channel, 1, window_size, window_size).contiguous()) |
| 53 | return window |
| 54 | |
| 55 | def ssim(img1, img2, window_size=11, size_average=True): |
| 56 | channel = img1.size(-3) |
no test coverage detected