我是机器学习的初学者,我正在尝试使用自己的数据集进行自己的对象检测。但是,如果对象用多边形边界标记会更实用。然而 tensorflow 对象检测 API 只能接受边界框。
那么是否可以修改 API 使其可以接受多边形标记的数据集?
我是机器学习的初学者,我正在尝试使用自己的数据集进行自己的对象检测。但是,如果对象用多边形边界标记会更实用。然而 tensorflow 对象检测 API 只能接受边界框。
那么是否可以修改 API 使其可以接受多边形标记的数据集?
对的,这是可能的。您必须提供训练集的目录。但推荐使用边界框,因为在推理期间,您会在检测到的对象周围获得边界框。您可以在 tensorflow.org 中看到一个示例。
对于标签,您可以使用LabelImg,它非常简单易用,也将提高检测精度。