Skip to content

Commit

Permalink
Merge pull request #533 from jahnvisahni31/patch-1
Browse files Browse the repository at this point in the history
[README Enhancement]: Colour Detection using DL
  • Loading branch information
abhisheks008 authored May 12, 2024
2 parents 226737c + 5113dac commit 2298542
Showing 1 changed file with 61 additions and 38 deletions.
99 changes: 61 additions & 38 deletions Colour Detection using DL/README.md
Original file line number Diff line number Diff line change
@@ -1,70 +1,93 @@
# Colour Detection using DL
## 🌈 Colour Detection using DL (#527)

## PROJECT TITLE

Colour Detection using DL
## 💡 Goal

## GOAL
In this project, we'll delve into the fascinating world of color detection using Deep Learning techniques. The goal is to accurately identify ten different colors from images.

To detect 10 different colours using DL techniques.
## 📊 Dataset Exploration

## DATASET
We've sourced our dataset from [Color Dataset for Color Recognition](https://www.kaggle.com/datasets/adikurniawan/color-dataset-for-color-recognition). This dataset contains 10 class of color, each color class contains 25 images.

The link for the dataset used in this project: https://www.kaggle.com/datasets/adikurniawan/color-dataset-for-color-recognition
## 🧾 Description
The project aims to create a color detection system using deep learning techniques. It involves identifying ten different colors from images. The dataset used for training and testing contains images labeled with various colors.

## DESCRIPTION
## 🧮 What I had done!
### 🔍 Data Preprocessing

This project aims to identify ten different colours from the given images.
Ensuring uniformity in image shapes and enhancing color contrast to extract meaningful features for our models.

## WHAT I HAD DONE
### 🧠 Model Selection

1. Data collection: From the link of the dataset given above.
2. Data preprocessing: Preprocessed the image in order to have all images in equal shape.
3. Model selection: Chose traditional CNN along with Image detection architecture VGG16 and ResNet15V2 for Image detection.
4. Comparative analysis: Compared the accuracy score of all the models.
We're employing the latest DL architectures for our color detection task:

## MODELS USED
1. **Convolutional Neural Network (CNN)**: Known for its process in image classification.
2. **VGG16**: A state-of-the-art architecture with deep convolutional layers.
3. **ResNet15V2**: Harnessing the power of residual networks for accurate color recognition.

1. CNN
2. VGG16
3. ResNet15V2
## Models Implemented

### CNN Model

## LIBRARIES NEEDED
**Description:**
- Utilized a traditional Convolutional Neural Network (CNN) architecture.
- CNNs are well-suited for image classification tasks due to their ability to learn spatial hierarchies.

The following libraries are required to run this project:
### VGG16 Model

**Description:**
- Employed the VGG16 architecture, known for its simplicity and effectiveness.
- VGG16 consists of 16 convolutional layers and is capable of capturing intricate features in images.

### ResNet50 Model

**Description:**
- Implemented the ResNet50 architecture, featuring residual connections.
- ResNet50's residual blocks mitigate the vanishing gradient problem, enabling training of deeper networks.

## Libraries needed
Here are the libraries needed for this project:

- numpy==1.24.3
- pandas==1.5.0
- matplotlib==3.6.0
- seaborn==0.11.2
- Pillow==8.4.0
- opencv-python==4.5.3.56
- keras==2.6.0
- tensorflow==2.6.0
- scikit-learn==0.24.2

## VISUALIZATION
![cnn](https://github.com/achrekarom12/DL-Simplified/assets/88442486/8336a43a-b9e0-4a74-a7dc-794c3b20f8ec)
![cnn 2](https://github.com/achrekarom12/DL-Simplified/assets/88442486/e68336fc-a0b3-488d-bce5-37b31fd12540)
### Exploratory data analysis results

### CNN Model
![CNN Model](https://github.com/abhisheks008/DL-Simplified/blob/main/Colour%20Detection%20using%20DL/Images/cnn%202.jpg)

### VGG16 Model
![VGG16 Model](https://github.com/abhisheks008/DL-Simplified/blob/main/Colour%20Detection%20using%20DL/Images/vgg16%202.jpg)

## EVALUATION METRICS
### ResNet50 Model
![ResNet16 Model](https://github.com/abhisheks008/DL-Simplified/blob/main/Colour%20Detection%20using%20DL/Images/res%202.jpg)

The evaluation metrics I used to assess the models:

- Accuracy
- Loss
## 📈 Performance Evaluation


## RESULTS
Results on Val dataset:
Let's visualize the accuracy scores of our models:

| Model | Accuracy | Loss |
|------------|----------|---------|
| CNN | 0.9 | 0.326 |
| VGG16 | 0.825 | 0.154 |
| ResNet50 | 0.875 | 0.987 |
| CNN | 90% | 0.326 |
| VGG16 | 82.5% | 0.154 |
| ResNet50 | 87.5% | 0.987 |


## 🚀 Conclusion

1. **CNN Model**: With an accuracy of 90% and a low loss of 0.326, it's the clear winner in color detection.
2. **VGG16 Model**: While respectable at 82.5% accuracy, it falls short compared to the CNN model.
3. **ResNet50 Model**: Despite its 87.5% accuracy, its higher loss indicates potential overfitting.

In conclusion, the CNN model emerges as the champion in accurately detecting colors from images.

## CONCLUSION
Based on results we can draw following conclusions:
1. CNN Model: The CNN model achieved the highest accuracy of 90% and a relatively low loss of 0.326 on the validation dataset. This indicates that the CNN model performed well in learning and predicting color classes.
2. VGG16 Model: The VGG16 model showed an accuracy of 82.5% and a loss of 0.154. While it performed reasonably well, it didn't outperform the CNN model in terms of accuracy.
3. ResNet50 Model: The ResNet50 model achieved an accuracy of 87.5% but had a relatively high loss of 0.987. This suggests that the model might be overfitting or facing challenges in generalizing to new data.
## Your Signature
Jahnvi sahni[https://github.com/jahnvisahni31]

0 comments on commit 2298542

Please sign in to comment.