https://github.com/fahrettinsolak/ai-object-recognition-classification-app-project

This project demonstrates image classification using Transfer Learning with the InceptionResNetV2 model. The model is pre-trained on the ImageNet dataset and can classify images into one of 1,000 categories. The application is built using Streamlit, allowing users to upload images and get predictions along with confidence scores.
https://github.com/fahrettinsolak/ai-object-recognition-classification-app-project

artificial-intelligence classification deep-learning image-processing jupyter-notebook machine-learning phyton

Last synced: 7 days ago
JSON representation

This project demonstrates image classification using Transfer Learning with the InceptionResNetV2 model. The model is pre-trained on the ImageNet dataset and can classify images into one of 1,000 categories. The application is built using Streamlit, allowing users to upload images and get predictions along with confidence scores.

• Host: GitHub
• URL: https://github.com/fahrettinsolak/ai-object-recognition-classification-app-project
• Owner: Fahrettinsolak
• Created: 2024-10-15T22:29:42.000Z (2 months ago)
• Default Branch: main
• Last Pushed: 2024-10-15T22:31:04.000Z (2 months ago)
• Last Synced: 2024-10-31T23:08:12.174Z (about 2 months ago)
• Topics: artificial-intelligence, classification, deep-learning, image-processing, jupyter-notebook, machine-learning, phyton
• Language: Jupyter Notebook
• Homepage:
• Size: 1.25 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# AI-Object-Recognition-Classification-App-Project

# Image Classification using Transfer Learning

![Before Prediction](app_interface.png)

This project demonstrates image classification using Transfer Learning with the **InceptionResNetV2** model. The model is pre-trained on the ImageNet dataset and can classify images into one of 1,000 categories. The application is built using **Streamlit**, allowing users to upload images and get predictions along with confidence scores.

## Project Overview

- **Model**: InceptionResNetV2 (pre-trained on ImageNet)

- **Framework**: TensorFlow Keras

- **Frontend**: Streamlit for a simple, interactive web interface

- **Input Size**: 299x299 pixels (required by InceptionResNetV2)

- **Categories**: This project includes a demo list of recognizable categories such as submarines, animals, vehicles, etc.

## How Transfer Learning Works

Transfer learning allows us to use pre-trained models on different datasets, reducing training time and achieving high accuracy. In this project, the InceptionResNetV2 model is used to classify user-uploaded images based on its knowledge from the ImageNet dataset.

### Key Features

- Upload an image through the web interface.

- Classify the image into one of 1,000 categories using the model.

- Display the predicted class and confidence score.

## Installation

1. Clone the repository:

```bash

https://github.com/Fahrettinsolak/AI-Object-Recognition-Classification-App-Project.git

```

2. Run the Streamlit app:

```bash

streamlit run main.py

```

## Usage

1. Upload an image through the "Choose an image..." option.

2. The model will process the image and display the predicted class and confidence score.

![After Prediction](result.png)

## Recognizable Categories (Demo)

This is a shortened list of recognizable categories from the ImageNet dataset:

Tench, Goldfish, White Shark, Tiger Shark, Hammerhead, Electric Ray, Stingray, Cock, Hen, Ostrich, Pelican, Penguin, Parrot, Goose, Bald Eagle, Koala, Chimpanzee, Orangutan, Tiger, Lion, Leopard, Elephant, Zebra, Camel, Kangaroo, Submarine, Fire Engine, Taxi, Race Car, Bicycle, Motorcycle, Sports Car, Airplane, Warplane, Helicopter.