A novel model to detect and categorize objects from images by using a hybrid machine learning model

Abstract

As humans, we can easily recognize and distinguish different features of objects in images due to our brain’s ability to unconsciously learn from a set of images. The objectives of this effort are to develop a model that is capable of identifying and categorizing objects that are present within images. We imported the dataset from Keras and loaded it using data loaders to achieve this. We then utilized various deep learning algorithms, such as visual geometry group (VGG)-16 and a simple net-random forest hybrid model, to classify the objects. After classification, the accuracy obtained by VGG16 and the hybrid model was 84.7% and 89.6%, respectively. Therefore, the proposed model successfully detects objects in images using a simple net as a feature extractor and a random forest for object classification, achieving better accuracy than VGG16.