maskrcnn-benchmark 代码详解之　image_list.py

最新推荐文章于 2021-08-31 21:24:23 发布

原创

最新推荐文章于 2021-08-31 21:24:23 发布 · 2.4k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#maskrcnn benchmark #ImageList.py #目标检测 #深度学习 #代码详解

本文详细解读maskrcnn-benchmark库中的image_list.py，主要介绍ImageList数据类型，它是用来存储图像列表并进行转换的工具。ImageList包含图像列表张量和图像大小列表。此外，代码还支持将不同格式的图像列表转换为ImageList，如Tensor或可迭代张量，并在必要时通过填充0调整图像尺寸。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

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 = self.tensors.to(*args, **kwargs)
        return ImageList(cast_tensor, self.image_sizes)