/object-detection

Multiclass bounding box regression task using VGG16 and fine tuning on CALTECH101 dataset

Primary LanguageJupyter Notebook

Bouding Box Regression

Overview

Here, I have trained two CNN architectures :

  • One predicts bounding boxes of a single class - Airplanes as shown below
    Notebook : bounding_box_regressor.ipynb
    Monoclass BB Regression

  • The other is fine tuned for predicting class labels as well as boxes for 5 different classes chosen at random from the CALTECH101 dataset.
    The link to the dataset, necessary preprocessing can be found out in multiclass_bounding_box.ipynb
    Multiclass BB Regression

Model Architecture

Regressor Architecture

The two output branches, one for classification and other for bounding box regression, improve the overall efficiency of the task cause being the use of a better loss function.

Tutorial : PyImageSearch