使用Tensorflow对象检测API进行自定义对象检测

时间:2018-10-13 14:40:01

标签: tensorflow object-detection-api mscoco

我想检测头盔和不戴头盔的人。.我使用ssd_mobilenet_v1_coco_11_06_2017模型,带1000个头盔图像和1000个不带头盔。

1)可以使用ssd_mobilenet_v1_coco_11_06_2017模型进行培训吗?如果没有,我可以使用哪种模型? (因为COCO只有90个班级和头盔不在该列表中)

有人可以帮助我清除我的疑问...