nanopyx.core.transform.padding

 1import numpy as np
 2
 3
 4def pad_w_zeros_2d(img, height, width):
 5    """
 6    Generate a 2D padded image with zeros.
 7
 8    Parameters:
 9        img (numpy.ndarray): The input image.
10        height (int): The desired height of the padded image.
11        width (int): The desired width of the padded image.
12
13    Returns:
14        numpy.ndarray: The padded image with zeros.
15    """
16    padded_img = np.zeros((height, width), dtype=img.dtype)
17    img_h, img_w = img.shape
18    padded_img[
19        (height - img_h) // 2 : (height - img_h) // 2 + img_h, (width - img_w) // 2 : (width - img_w) // 2 + img_w
20    ] = img
21
22    return padded_img