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?

  • $\begingroup$ You want to train an object detection model on 25 images? How would that work? $\endgroup$ – theblackips Oct 3 '19 at 11:18
  • $\begingroup$ Since any object detection network like yolo or faster rcnn would take more time to train on many images in a cpu, I want to build yolo or faster rcnn network architecture, train on only 20 images(with 2 classes, 2 kinds of flowers) and validate on 5 images with few epochs.Randomly initialize weights from layer 1 and not to use pre trained model and weights derived from training with imagenet or cats and dogs data set. This exercise is for learning and understanding the architecture. $\endgroup$ – Malathi Oct 3 '19 at 16:23

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.