可以使用多边形数据标注来执行TensorFlow对象检测吗?

我的问题并不完全是使用多边形、圆形或线条来标注数据,而是如何使用这些标注的数据生成一个”.tfrecord“文件并执行对象检测。我看到的教程使用的是矩形标注,像这些:泰勒·斯威夫特检测 浣熊检测

如果我想要检测的对象(管道)彼此之间不是太靠近,这对我来说将是一个很好的解决方案。

在PASCAL VOC格式中绘制的矩形示例:

<bndbox>    <xmin>82</xmin>    <xmax>172</xmax>    <ymin>108</ymin>    <ymax>146</ymax></bndbox>

是否有办法添加一个“掩码”来突出显示这个边界框的某些部分?

如果有任何不清楚的地方,请告诉我。


回答:

如果你要检测的对象彼此非常靠近,你可以选择实例分割而不是对象检测,在这种情况下,你可以使用多边形来生成掩码和边界框来训练模型。

考虑使用这个展示良好且易于使用的存储库来进行mask-rcnn(一种实例分割):

https://github.com/matterport/Mask_RCNN

查看这个来获取轻量级的mask-rcnn

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中创建了一个多类分类项目。该项目可以对…

发表回复

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