如何训练条件数据集?

时间:2019-07-23 17:21:09

标签: machine-learning artificial-intelligence object-detection

我有一个对象检测项目,正在使用Google的Inception Model进行研究。该模型应该能够检测出没有佩戴头盔的人,所以我主要关心的是,我是否可以对其进行培训,以便能够(如果并且仅当他/她坐在摩托车上时)能够检测到人员是否戴着头盔?如果是,请帮助我。

1 个答案:

答案 0 :(得分:0)

可以。

方法1-将其视为来自的多标签分类。标记“摩托车”和“头盔”类的数据集。这是一本不错的书-https://towardsdatascience.com/journey-to-the-center-of-multi-label-classification-384c40229bff

方法2-考虑两步分类。您将需要两个模型。模型1将检测摩托车。如果摩托车检测到,则将数据传递给将检测头盔的Model 2。