Custom object detection using Tensorflow Object Detection API

Question

I want to detect helmet and person not wearing helmet.. I use ssd_mobilenet_v1_coco_11_06_2017 model with 1000 helmet images and 1000 without helmet.

1) Is that ok to use ssd_mobilenet_v1_coco_11_06_2017 model for training? If not then what model I can use? (because COCO have only 90 classes and helmet not in that list)

can somebody help me to clear my doubt...

score 1 · Accepted Answer · answered Oct 17 '18 at 06:39

1

Refer following links to achieve this

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

answered Oct 17 '18 at 06:39

Srinivas Bringu

452
3
9

Thanks for your help – Malathi K Oct 23 '18 at 11:39

Custom object detection using Tensorflow Object Detection API

1 Answers1