This repository contains code for object recognition using ResNet50, a pre-trained convolutional neural network. The code is written in Python and uses various libraries such as NumPy, Pandas, PIL, Matplotlib, and OpenCV.
The dataset used in this project is taken from the CIFAR-10 dataset. It contains images of various objects such as airplanes, automobiles, birds, cats, deer, dogs, frogs, horses, ships, and trucks. The labels for these images are available in the trainLabels.csv
file.
The code is available in the object_recognition_using_resnet50.ipynb
Jupyter Notebook file. It contains the following sections:
- Loading the dataset
- Displaying sample images
- Converting images to NumPy arrays
- Splitting the dataset into training and testing sets
- Scaling the data
- Building the neural network using ResNet50
- Compiling and training the neural network
To run this code, you need to have the following libraries installed:
- NumPy
- Pandas
- PIL
- Matplotlib
- OpenCV
- TensorFlow
The dataset used in this project is taken from the CIFAR-10 dataset.
The ResNet50 model used in this project is available in the Keras Applications module.