tensorflow / tensorflow/models
How to retrain existing SSD_Mobilenet_MSCOCO model for a new class label
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Dear all,
I run the trained model (ssd_mobilenet_v1_coco_11_06_2017) to detect bounding box on my dataset, but this trained model is not performing well because the main object that I am detecting is not in trained model.
Here is example that I am following:
https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb
First Question: How can I retrain the existing ssd_mobilenet_v1_coco_11_06_2017 model with one addition class. For the new class, I generate boundingbox annotation in xml format.
Second Question: How can I train a new model only with my own dataset which has only two classes. If possible share any demo or code or tutorial.
Thank you.
-Humayun
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.