1

我想对原始图像及其分段蒙版进行增强,但我总是会遇到一些错误,例如:

  1. ValueError:使用序列设置数组元素
  2. 无法将排名第 3 的图像扩展为排名第 4 的图片

这是我的代码:

data_gen_args = dict(rotation_range=5)
image_datagen = ImageDataGenerator(**data_gen_args)
mask_datagen = ImageDataGenerator(**data_gen_args)

seed = 1
images= []
masks= []

for image in glob.glob('../data/images/*.jpg'):
    images.append(cv.imread(image))
for image in glob.glob('../data/masks/*.jpg'):
    masks.append(cv.imread(image))

images= np.array(images)
masks= np.array(masks)

for image in images:
    image_datagen.fit(image[np.newaxis], augment=True, seed=seed)

for image in masks:
    mask_datagen.fit(image[np.newaxis], augment=True, seed=seed)

image_generator = image_datagen.flow(images, seed=seed)
mask_generator = mask_datagen.flow(masks, seed=seed)

train_generator = zip(image_generator, mask_generator)

id = 0
for images, masks in train_generator:
    for image, mask in zip(images, masks):
        cv.imwrite('../data/aug_images/images/' + str(id), image)
        cv.imwrite('../data/aug_images/masks/' + str(id), mask)
        id += 1

有人可以纠正我哪里出错了吗?谢谢你的帮助。

编辑:我有一组图像,它是蒙版。我想为他们两个获得相同的增强/转换。

4

0 回答 0