A Streamlit app that uses the ResNet50 pre-trained model to classify objects from images.
A residual neural network (ResNet) is an artificial neural network (ANN). It is a gateless or open-gated variant of the HighwayNet, the first working very deep feedforward neural network with hundreds of layers, much deeper than previous neural networks.
ResNet-50 is a convolutional neural network that is 50 layers deep. You can load a pretrained version of the network trained on more than a million images from the ImageNet database. The pretrained network can classify images into 1000 object categories, such as keyboard, mouse, pencil, and many animals.
To run a demo do the following:
- Clone the repository.
- Install the requirements from the requirements.txt file:
pip install -r requirements.txt
- Then from your command line run:
streamlit run streamlit_app.py
Then you can view the site on your local server.
The app was made and deployed with streamlit and streamlit cloud. Streamlit is an open source app framework in Python language. It helps us create web apps for data science and machine learning in a short time. It is compatible with major Python libraries such as scikit-learn, Keras, PyTorch, SymPy(latex), NumPy, pandas, Matplotlib etc.
The live link can be seen here: https://share.streamlit.io/nneji123/deep-learning-object-recognition-with-resnet50/main