我正在使用ImageGenerator
的功能tf2.4
来增强植物图像,但是,生成的图像由外部图像边界的像素点填充,这与 的 模式{"constant", "nearest", "reflect" or "wrap"}
有关ImageGenerator
,在我的情况下不需要。关于ImageGenerator
API,没有禁用这些模式的选项。
我得到的生成图像的样本ImageGenerator
并给出了参数。
代码:
generator = tf.keras.preprocessing.image.ImageDataGenerator(vertical_flip=45)
data = generator.flow_from_directory(
data_dir, target_size=(IMG_SIZE, IMG_SIZE), color_mode='rgb',
class_mode='sparse',
batch_size=batch_size,
shuffle=True,
save_to_dir=OUTPUT_DIR,
save_prefix='augmented',
save_format='png',
follow_links=False,
subset=None,
interpolation='nearest'
)
结果:
红色画线的内部区域是我提到的将像素点填充到图像边界之外的区域。
我很想知道是否有任何选项可以解决我的目标?