调整可变大小的图像大小以进行对象检测

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (62)

我正在使用KITTI的物体检测数据集来训练更快的R-CNN和ResNet 101,预先训练在MS COCO上。KITTI图像“大部分”尺寸为375x1242

batch_size: 1 keep_aspect_ratio_resizer就像下面一样,因为它是由TensorFlow本身提出的。

min_dimension: 600
max_dimension: 1987

但现在我想使用batch_size:5,但我不断遇到尺寸不匹配错误。因为某些图像的尺寸略小,如370x1224等。

我找不到一般的keep_aspect_ratio_resizer价值观。我根据我在错误消息中看到的值尝试了以下值,但我无法使所有图像真正具有相同的大小

min_dimension: 600
max_dimension: 1985

min_dimension: 599
max_dimension: 1985
提问于
用户回答回答于

您可以自己调整图像大小以解决此处的舍入错误,或者您可以尝试将相同大小的项目组合成批处理。

假设您有一个图像生成器,并且图像具有大小属性​​,您可以创建一个生成器,它将生成大小相同的图像,如下所示:

from itertools import groupby
from collections import deque, defaultdict


def same_size_batches(images, batch_size=5):
    image_cache = defaultdict(deque)

    # We assume the image object has a size parameter we can group by
    for size, images in groupby(images, key=lambda image: image.size):
        for image in images:
            image_cache[size].append(image)

            # Everytime our batch gets big enough, yield it and reset
            if len(image_cache[size]) == batch_size:
                yield iter(image_cache[size])
                image_cache[size].clear()

这里的主要部分是按相同键对连续项进行分组的组,并返回该键和与该键匹配的项的生成器。在我们的例子中,关键是图像的大小。

然后我们保留相同大小的项目的缓存,并且每当其中一个大小达到我们期望的批量大小时,我们就为该批次生成一个生成器。

我们可以使用具有所需大小参数的伪图像对象来演示这一点:

import random


class FakeImage(object):
    def __init__(self, _id):
        self.id = _id
        self.size = (370, 1224) if random.random() < 0.25 else (375, 1242)

    def __repr__(self):
        return "<Image {} {}>".format(self.id, self.size)


images = (FakeImage(_id) for _id in range(100))
for batch in same_size_batches(images, batch_size=5):
    print(list(batch))

这导致类似于:

[<Image 0 (375, 1242)>, <Image 2 (375, 1242)>, <Image 3 (375, 1242)>, <Image 4 (375, 1242)>, <Image 6 (375, 1242)>]
[<Image 7 (375, 1242)>, <Image 8 (375, 1242)>, <Image 9 (375, 1242)>, <Image 10 (375, 1242)>, <Image 12 (375, 1242)>]
[<Image 1 (370, 1224)>, <Image 5 (370, 1224)>, <Image 11 (370, 1224)>, <Image 14 (370, 1224)>, <Image 16 (370, 1224)>]
[<Image 13 (375, 1242)>, <Image 15 (375, 1242)>, <Image 18 (375, 1242)>, <Image 19 (375, 1242)>, <Image 20 (375, 1242)>]
[<Image 21 (375, 1242)>, <Image 23 (375, 1242)>, <Image 24 (375, 1242)>, <Image 25 (375, 1242)>, <Image 26 (375, 1242)>]
...

如果我们没有足够填充批量大小的块,我们不能保证生成所有图像,但如果输入是无限生成器,则这不是问题。

扫码关注云+社区

领取腾讯云代金券