make kernel of same type as image to avoid mismatch issues (#9932)

This commit is contained in:
Jodh Singh
2025-09-18 19:51:16 -04:00
committed by GitHub
parent 8d6653fca6
commit 1ea8c54064

View File

@@ -233,6 +233,7 @@ class Sharpen:
kernel_size = sharpen_radius * 2 + 1
kernel = gaussian_kernel(kernel_size, sigma, device=image.device) * -(alpha*10)
kernel = kernel.to(dtype=image.dtype)
center = kernel_size // 2
kernel[center, center] = kernel[center, center] - kernel.sum() + 1.0
kernel = kernel.repeat(channels, 1, 1).unsqueeze(1)