models icon indicating copy to clipboard operation
models copied to clipboard

How to to train custom ssd mobilenet from scratch

Open yieniggu opened this issue 5 years ago • 2 comments

How can I retrain a ssd-mobilenet-v2 from the tensorflow object detection model zoo without transfer learning. I mean every weight and not just the last layer.

Do I have to build the network architecture and the script for training or can I make some minor change to the .config file or the train.py script specified in the docs on training a custom object detector.

Will this approach improve/worsen the network accuracy/loss?

Thanks in advance for any help.

yieniggu avatar Mar 30 '20 11:03 yieniggu

Also looking for a way to do this, for any model using the object detection model zoo pipeline.

I seriously doubt it will improve accuracy for my use-case, but I need to train a reinitialised model from scratch to compare against.

Is there a way, following the steps outlined in the Object Detection API tutorial, to re-initialise the parameters of the model and train from scratch?

Wheest avatar Dec 21 '20 22:12 Wheest

How can I retrain a ssd-mobilenet-v2 from the tensorflow object detection model zoo without transfer learning. I mean every weight and not just the last layer.

Do I have to build the network architecture and the script for training or can I make some minor change to the .config file or the train.py script specified in the docs on training a custom object detector.

Will this approach improve/worsen the network accuracy/loss?

Thanks in advance for any help.

Did you find the answer? if you could help me in this regards?

Annieliaquat avatar Oct 24 '23 07:10 Annieliaquat