tensorflow / tensorflow/models
Tensorflow 2 Object Detection API - Training MobileNet v2 from scratch
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
I initially followed this tutorial to fine tune a SSD MobileNet v2 320x320 model from the TF2 model zoo.
However, I want to try training this MobileNetV2 model with a depth multiplier of 0.75. To do this, I understand that I need to retrain the MobileNetV2 model 'from scratch'. So I downloaded the COCO2017 dataset, with the intent to use this config file to train the model. This config mentions that it was trained from an ImageNet checkpoint, so I retrieved a MobileNet checkpoint trained on ImageNet checkpoint with dm=0.75 from here. However, the checkpoint files in this seem to be a different format to the checkpoints provided with the pre trained models on the model zoo? Eg, the model zoo models provide checkpoint, ckpt-0.data-00000-of-00001 and ckpt-0.index, whereas the imagnet checkpoint is missing the 'checkpoint' file.
When I try to train the model on any dataset using the downloaded config and ImageNet checkpoint, I get an error that the checkpoint is expected to be an object-based checkpoint?
I've uploaded my pipeline.config (I had to change the extension to .txt to upload it), from when I tried to train the model on a simple dataset with only one object class.
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.