I tried using DAG for testing the CNN, but it does not allow two output layers (classification head and regression head). How should I proceed? Any suggestions?
How to combine trained classification and regression heads of a CNN during testing?
3 views (last 30 days)
Show older comments
I am building a 3 class object detector using CNN. Since my classes are less, I am using a smaller CNN called VGG-M which is pre-trained on ImageNet dataset. I am using transfer learning to train the model to act as a 3 class classifier as well as regressor. I trained the model for classification and regression by attaching the classification and regression heads respectively after the fully connected layers of the original model. However, during testing, I want my model should give me class of the object in the image as well as its bounding box. In Matlab 2017b, I am not able to find a suitable way of doing it. Experts are requested to suggest a solution. Thank you in advance!
3 Comments
Krishna Bindumadhavan
on 16 Mar 2018
Could you try designing an object detector using Faster R-CNN? You can refer to the example here to get started.
Answers (0)
See Also
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!