标签: tensorflow object-detection-api mscoco
我想检测头盔和不戴头盔的人。.我使用ssd_mobilenet_v1_coco_11_06_2017模型,带1000个头盔图像和1000个不带头盔。
1)可以使用ssd_mobilenet_v1_coco_11_06_2017模型进行培训吗?如果没有,我可以使用哪种模型? (因为COCO只有90个班级和头盔不在该列表中)
有人可以帮助我清除我的疑问...
答案 0 :(得分:0)
请参阅以下链接以实现此目的
https://pythonprogramming.net/training-custom-objects-tensorflow-object-detection-api-tutorial/
https://towardsdatascience.com/building-a-toy-detector-with-tensorflow-object-detection-api-63c0fdf2ac95
https://medium.com/@dana.yu/training-a-custom-object-detection-model-41093ddc5797