我想检测头盔和不戴头盔的人。我使用 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 个班级,头盔不在该列表中)
有人可以帮我清除我的疑问...
请参阅以下链接以实现此目的
https://pythonprogramming.net/training-custom-objects-tensorflow-object-detection-api-tutorial/
https://medium.com/@dana.yu/training-a-custom-object-detection-model-41093ddc5797