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...

Srinivas Bringu · Accepted Answer · 2018-10-17 06:39:26Z

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, 2018 at 6:39

Srinivas Bringu

4523 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Malathi K Over a year ago

Thanks for your help

Collectives™ on Stack Overflow

Custom object detection using Tensorflow Object Detection API

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related