Fix padding to bottom-right

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):