使用 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 类 而头盔不在该列表中)
谁能帮我解开疑惑...
我想检测头盔和不戴头盔的人。我使用 ssd_mobilenet_v1_coco_11_06_2017 模型,其中有 1000 个头盔图像和 1000 个不戴头盔的图像。
1) 用ssd_mobilenet_v1_coco_11_06_2017模型训练可以吗?如果不是那么我可以使用什么型号? (因为 COCO 只有 90 类 而头盔不在该列表中)
谁能帮我解开疑惑...