如何在 TensorFlow Lite Model Maker 中指定增强？

Question

我正在尝试在 TensorFlow Lite Model Maker 的帮助下按照本教程制作图像分类器：https://www.tensorflow.org/lite/tutorials/model_maker_image_classification。

当我使用默认值时一切正常。

我要做的是配置超参数，更准确地说是增强超参数：use_augmentation。 https://www.tensorflow.org/lite/tutorials/model_maker_image_classification#change_the_training_hyperparameters

文档说：

use_augmentation：布尔值，使用数据增强进行预处理。默认为假。

现在我的问题是使用什么数据增强？并且可以配置这种增强功能吗？就像我想使用horizo​​ntal_flip，指定旋转和缩放范围等。

提前致谢！

Answer 1

通过转到tflite_model_maker.img 的源代码并遵循use_augmentation 参数，我来到了here，在那里完成了增强。如果模型正在训练，则增强似乎是随机裁剪和翻转。在评估时，显然没有进行随机裁剪或翻转，而只是进行了中心裁剪。