使用 ML 为对象检测准备自定义数据集

数据挖掘 机器学习 神经网络 深度学习 张量流 物体检测
2022-02-25 07:45:46

使用 ML 寻求单类对象检测模型的清晰度。为此,我准备了一个自定义数据库,最多可容纳 400 张图像,这些图像分为 80%-20% 作为训练和测试数据集。这些是仅顶视图图像。数据收集遵循此处提供的基本准则。

在此处输入图像描述

现在的目标是在以下情况下检测斑马线。

在此处输入图像描述

模型在准确性方面失败。虽然它识别了类,但在某些情况下(更大程度)无法正确定位。理想的结果是蓝色,而模型抛出红色。 在此处输入图像描述

需要对训练数据集进行哪些更改才能纠正此问题?

EDIT1:- Tensorflow 对象检测 API 用于此任务。检测精度达到90%以上。期待解决本地化问题的建议。

EDIT2:- 这里的插图仅用于概述问题。我所有的图片都是真实世界的图片。

1个回答

我有以下建议:

  • 你的数据集很小,你应该增加样本量。您还必须使用数据增强方法来提高性能,强烈推荐平移、方向和模糊。

  • 对于您的任务,我想最流行的方法是YOLO论文中讨论的方法,如果您想要进行对象检测而不仅仅是定位和识别以获得良好的性能,强烈建议使用该方法。

  • 我还建议您为您的任务提供真实数据。第一行的图像是人工制作的。您必须提供分类器将要面对的真实数据。如果您想获得良好的性能,您的测试时间的分布应该类似于验证和训练时间。