maskrcnn-benchmark 代码详解之 image_list.py

前言

image_list.py主要是用于保存图像列表的数据类型以及将其他保存图像列表的数据类型转换成maskrcnn benchmark所规定的ImageList类型。ImageList的属性包含一个保存图像列表的张量还有一个保存图像大小的列表，其代码为：

class ImageList(object):
    """
    Structure that holds a list of images (of possibly
    varying sizes) as a single tensor.
    This works by padding the images to the same size,
    and storing in a field the original sizes of each image
    用于保存图片（大小可以不同)列表并使之成为一个单一张量的数据结构，
    通过扩充图像到相同的大小来保存原图像的大小
    """

    def __init__(self, tensors, image_sizes):
        """
        Arguments:
            tensors (tensor)
            image_sizes (list[tuple[int, int]])
        """
        # 该数据类型有两个属性，保存图像数据的tensors，和图像各自的大小image_sizes
        self.tensors = tensors
        self.image_sizes = image_sizes

    # 实现转换功能，比如todevice todetype等等
    def to(self, *args, **kwargs):
        cast_tensor &#