谈运动训练中的图像增强

Question

我在做图像分类问题。我想要像调整大小，翻转，剪裁，随机对比度等图像增强，我想知道我应该先增加图像并将它们保存到of记录中使用，还是应该创建原始图像的tfrecord，并在训练期间进行增强？它会产生巨大的计算资源差异吗？

谢谢

Answer 1

不，正如您在本例中所看到的，您可以在您使用的模型的pipeline.config中添加您想要的任何内容。例如，可以将此部分添加到ssd_mobile_net的配置中。

train_config {
  batch_size: 64
  data_augmentation_options {
    random_horizontal_flip {
    }
  }
  data_augmentation_options {
    random_vertical_flip{
    }
  }
  data_augmentation_options {
    random_rotation90{
    }
  }
  data_augmentation_options {
    random_pad_image{
    }
  }

要可视化每个选项对图片所做的操作，可以使用以下代码。增强