CODEWITHSUNDEEP
tensorflowobject-detectionensemble-learning
pierre.ecarlat
pierre.ecarlat

Reputation: 43

Object Detection API - How to create an Ensemble of trainings?

I already created an Ensemble for classification (average -or so- of predictions per images), or for Semantic Segmentation (average -or so- of predictions per pixels), but I don't really know how to proceed for Object Detection.. My guess would be to extract all the region proposals of all my networks, then to run my classifiers on the X best of them, and finally to average the predictions for all the bounding boxes. But how should I do that with architectures following the Object Detection API?

I guess the regions proposals can be extracted using extract_proposal_features, and then reinserted to the model, but the only way I see to do that would be to create a complete new model with its own predict method etc, dealing will all the models of my Ensemble. Am I missing an other obvious / simpler method?

Upvotes: 4

Views: 1339

Answers (1)

Jonathan Huang
Jonathan Huang

Reputation: 1558

That's the basic idea, yes (the Resnet paper has a good explanation of how this is done for Faster R-CNN). Unfortunately we haven't released code to automate this ensembling process (and don't have any plans to). It's possible of course; you will have to manually set this up yourself.

Upvotes: 2

Related Questions