How can i use the validation set in object detection using deep learning?

1 次查看(过去 30 天)
I use faster-RCNN using pre-trained model(ResNet-50) for object detection.
I'm fine-tunning the last layers using a custom dataset.
As far as I know, when you have a small dataset k-fold cross validation is the right technique for evaluation.
But, this example does not seem to go through a validation procedure. Is there something I'm missing?

回答(1 个)

Divya Gaddipati
Divya Gaddipati 2019-7-17
The trainFasterRCNNObjectDetector does not support the “ValidationData”, “ValidationFrequency” and “ValidationPatience” options as of now. Hence, it is not possible to directly feed the validation data to the trainingOptions parameter.
Refer to this documentation for more information on how to use trainFasterRCNNObjectDetector function and the Tips section at the end of the page:
Possible workaround for the k-fold validation could be to run a loop “k” times.
  • In each loop, split the train set randomly into two sets, where one set will be used for training and the other for validation. You can use crossvalind or cvpartition functions for this purpose.
  • Use this new train set for training the model and after the training is done, use the validation set to evaluate the model.
This above step is done “k” times.
For more information of k-fold cross validation, you can refer to the following links:

类别

Help CenterFile Exchange 中查找有关 Gaussian Process Regression 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by