以不同方向训练具有相同图像的模型

Posted

技术标签:

【中文标题】以不同方向训练具有相同图像的模型【英文标题】:Train Model with same image in diferents orientation 【发布时间】:2018-08-15 18:03:08 【问题描述】:

用相同的图像训练模型是个好主意,但方向不同?我有一小部分图像用于训练,这就是我试图涵盖所有移动相机图库用户场景的原因。

例如图片:example.png 有3个副本; example90.png、example180.png 和 example.270.png 及其不同的旋转。并且还具有不同的背景颜色、阴影等。

顺便说一句,我的测试是识别动物的类型。

这是个好主意吗?

【问题讨论】:

取决于你想做什么。但总的来说,我会说是的,这是一种数据增强形式。如果您想要一个准确的答案,请添加更多详细信息,因为您的问题太宽泛了。 抱歉,我只是担心训练模型专注于同一组图像,并且模型只会积极识别与训练集非常相似的图像 【参考方案1】:

如果您将 Core ML 与 Vision 框架一起使用(您可能应该这样做),Vision 会自动旋转图像,以便“向上”真正向上。在这种情况下,用户在拍照时如何拿着相机并不重要(假设图片仍然有描述其方向的 EXIF 数据)。

【讨论】:

以上是关于以不同方向训练具有相同图像的模型的主要内容,如果未能解决你的问题,请参考以下文章

在 keras 的输出阶段组合多个预训练模型

基于图像和文本特征的 TensorFlow 训练模型,具有多类输出

具有更大输入图像尺寸的 Caffe 预训练模型

如何训练具有可变输出大小的对象检测模型?

具有 LSTM 网络的连体模型无法使用 tensorflow 进行训练

全卷积网络 - 训练数据中不同大小的图像