Open AccessIdentification of Microcontroller Based Objects Using Image Classification in Python
Author(s) -
Zenith Nandy
Publication year - 2021
Publication title -
international journal for research in applied science and engineering technology
Language(s) - English
Resource type - Journals
ISSN - 2321-9653
DOI - 10.22214/ijraset.2021.37920
Subject(s) - python (programming language) , softmax function , computer science , scripting language , artificial intelligence , artificial neural network , convolutional neural network , microcontroller , pattern recognition (psychology) , classifier (uml) , deep learning , programming language , embedded system
In this paper, I built an AI model using deep learning, which identifies whether a given image is of an Arduino, a Beaglebone Black or a Jetson Nano. The identification of the object is based on prediction. The model is trained using 300 to 350 datasets of each category and is tested multiple times using different images at different angles, background colour and size. After multiple testing, the model is found to have 95 percent accuracy. Model used is Sequential and uses Convolution Neural Network (CNN) as its architecture. The activation function of each layer is RELU and for the output layer is Softmax. The output is a prediction and hence it is of probability type. This is a type of an application based project. The entire scripting is done using Python 3 programming language. Keywords: image classification, microcontroller boards, python, AI, deep learning, neural network