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!