重新训练TensorFlow对象检测API以识别特定车型 — 如何准备训练数据?

我是对象检测的新手,正在尝试重新训练TensorFlow中的对象检测API以在照片中识别特定车型。在准备自己的训练数据以重新训练模型时,除了绘制边界框等操作之外,我的疑问是,是否还需要在训练数据中准备负面示例(即不是我感兴趣的车型的汽车)以达到良好的性能?

我已经阅读了一些教程,这些教程通常给出检测一种对象的示例,并且他们准备的训练数据只标记了那种类型的标签。我在想,既然模型首先会提出一些感兴趣的区域,然后尝试对这些区域进行分类,如果我想从照片中检测非常具体的东西,是否也应该准备负面示例。

我正在重新训练基于faster_rcnn的模型。感谢您的帮助。


回答:

是的,为了获得更好的性能,您也需要负面示例。看起来您在考虑使用迁移学习来训练一个预训练的faster_rcnn模型,以添加一个新的类别来识别您的定制车型。您应该从相同数量的正面和负面示例(带有标记边界框的图像)开始。除了您目标车型之外,您还需要准备几种负面类别的示例(例如,负面车型1,负面车型2,负面车型3)。

您可以在我的GitHub仓库的数据文件夹中查看用于迁移学习的训练数据示例,这些示例包括一个正面类别和多个负面类别:PSV Detector Github

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注