Training object detection models from the scratch without using pre-trained models and weights

Question

Training object detection models from the scratch without using pre-trained models and weights

Malathi

2022年3月8日 05:04

I have a data set of 25 images. I wish to run Faster RCNN or yolov3 object detection models on this images.I want to create my custom trained model and get weights after running say 10 epochs. Later I can save these weights and use that for prediction. Build a model, train on my image data set and get weights. Is it possible?

Topic object-detection jupyter keras python

Category Data Science

Pranshu Mishra · Accepted Answer · 2020年6月8日 18:08

Maybe you could first train the model on a large sized flower dataset, that way the conv layer will be optimised to detect flowers and then do transfer learning on that model using your custom dataset freezing all but the last layer. I am not sure though, 25 images seem way too less. Try data augmentation on those images like horizontal and vertical flips, noise, shear etc.

Training object detection models from the scratch without using pre-trained models and weights

About