使用 Tensorflow 对象检测的自定义对象检测 API

Custom object detection using Tensorflow Object Detection API

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

1) 用ssd_mobilenet_v1_coco_11_06_2017模型训练可以吗?如果不是那么我可以使用什么型号? (因为 COCO 只有 90 类 而头盔不在该列表中)

谁能帮我解开疑惑...

参考以下链接来实现这个

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