标签: machine-learning dataset object-detection
我对机器学习很新,并尝试编写自己的神经网络进行对象检测。我已经下载了Open Images dataset,但我只需要检测一小组类。因此,我想减小数据集的大小,因为我不希望生成的模型检测到与我的特定任务无关的披萨或其他对象。
我不确定是否应该只删除我不需要的类的标签,也不删除所有使用此类标记的图像。实际上我只需要3个类,但数据集包含600个带有边界框的不同类。而且我真的不知道应该如何修改数据集,以便训练我的模型变得合理。