Automated PR - 2026-09-22

This commit is contained in:
github-actions[bot]
2026-09-22 08:20:29 +00:00
parent dfb2786749
commit 35364168d9
+2 -3
View File
@@ -10,7 +10,6 @@ from kornia.geometry.transform.pyramid import (
build_pyramid,
find_next_powerof_two,
is_powerof_two,
pad,
)
from torch import Tensor
@@ -28,7 +27,7 @@ def _pad_for_laplacian(image: torch.Tensor) -> tuple[torch.Tensor, tuple[int, in
if not (is_powerof_two(h) and is_powerof_two(w)):
pad_right = find_next_powerof_two(w) - w
pad_down = find_next_powerof_two(h) - h
image = pad(image, (0, pad_right, 0, pad_down), "reflect")
image = F.pad(image, (0, pad_right, 0, pad_down), mode="reflect")
return image, (pad_right, pad_down)
@@ -41,7 +40,7 @@ def _gaussian_pyramid(
h, w = images.shape[2], images.shape[3]
if not (is_powerof_two(w) and is_powerof_two(h)):
padding = (0, find_next_powerof_two(w) - w, 0, find_next_powerof_two(h) - h)
images = pad(images, padding, border_type)
images = F.pad(images, padding, mode=border_type)
return build_pyramid(images, max_level, border_type, align_corners)