python - 如何使用tensorflow对象检测API提高对象检测的精度?

标签 python api tensorflow object-detection

我正在应用tensorflow对象检测api来构建模型来检测单个对象。我自己的数据集包含 2150 个用于训练的图像和 540 个用于测试的图像。

所有图像的尺寸均为 1920(宽)x 1080(高)。每幅图像中的物体都非常小,大约为 55 x 15。此外,还有很多噪点。这是正确识别的我的数据集的示例图像(大多数图像无法识别):

enter image description here

我已经尝试了 detetion_model_zoo 中的所有型号,他们都没有给我满意的结果。 训练后 mAP 非常低。精度仅为0.3左右。以下是使用ssd_inception_v2_coco的评估:

enter image description here

对改进模型有什么建议吗?非常感谢!

最佳答案

我会推荐以下内容:

1) 将图像分成更小的裁剪,以便感兴趣的对象最终变得更大一些。您可以仅在推理时执行此操作,也可以同时对裁剪进行训练和推理。如果某些裁剪不包含任何有趣的东西,请不要担心。

2) 使用更强大的模型,例如带有 resnet 或 inception resnet 的更快的 r-cnn。

3) 弄清楚人类的表现是什么,以了解机器学习系统的性能上限可能是多少。

关于python - 如何使用tensorflow对象检测API提高对象检测的精度?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46161027/

相关文章:

ruby - 如何使用 Ruby 在 Hadoop HDFS 中写入和读取文件?

php - 我的 Twitter php 脚本停止工作了!

python - 在 tensorflow 计算机视觉建模中在哪里定义批量大小?

python - Tensorflow:从任意长度的复杂张量中提取连续的补丁

java - 使用 Jython/Python 中的 Open Trip Planner 脚本编写的出发地目的地矩阵

python - 如何在Python中访问嵌套函数的变量?

python - 处理 Word 文档的最佳方式

python - 和小于等于k的最长子数组长度

android - 注册过程中通过来自 api 的消息进行手机号码验证,例如 "viber"for android

machine-learning - 几个批处理的 TensorFlow 平均梯度