diff --git a/dan/utils.py b/dan/utils.py index 9eddb9c561ceac19baab53cf78c33cf768e1e917..19a3088cbf5dc85e3784ca3cbdf73478c12a9e1e 100644 --- a/dan/utils.py +++ b/dan/utils.py @@ -26,11 +26,11 @@ def pad_sequences_1D(data, padding_value): return padded_data -def pad_images(data): +def pad_images(images): """ - Pad the images so that they are in the middle of the large padded image (tb-lr mode). - :param data: List of numpy arrays. - :return padded_data: A tensor containing all the padded images. + Pad the images so that they are at the top left of the large padded image. + :param images: List of images as torch tensors. + :return padded_images: A tensor containing all the padded images. """ longest_x = max([x.shape[1] for x in data]) longest_y = max([x.shape[2] for x in data]) @@ -46,7 +46,7 @@ def pad_images(data): top : padded_data.shape[2] - bottom, left : padded_data.shape[3] - right, ] = image - return padded_data + return padded_images def read_image(path):