UppuluriKalyani / ML-Nexus

ML Nexus is an open-source collection of machine learning projects, covering topics like neural networks, computer vision, and NLP. Whether you're a beginner or expert, contribute, collaborate, and grow together in the world of AI. Join us to shape the future of machine learning!
https://discord.gg/fy8MQkCh
MIT License
26 stars 42 forks source link

Children vs Adults Classification using DL #203

Open pavitraag opened 2 hours ago

pavitraag commented 2 hours ago

Goal

The main goal of this project is to develop a classification model capable of accurately distinguishing between images of children and adults. The purpose is to explore the performance of different deep learning models specifically tailored for this classification task.

Dataset

The dataset used for this project can be found here. It consists of a collection of labeled images containing children and adults.

Description

This project aims to build a classification model that can analyze facial features and classify images as either children or adults. By leveraging deep learning models, the project seeks to achieve accurate and reliable classification results.

What I had done!

  1. Data collection - Gathered a diverse dataset of images containing children and adults.

  2. Data preprocessing - Performed essential preprocessing steps, including resizing, normalization, and augmentation, to prepare the data for training.

  3. Model selection - Chose popular deep learning models, including VGG19, ResNet50, InceptionV3, and MobileNetV2, for the classification task.

  4. Model training - Trained each model using the labelled dataset and appropriate training configurations.

  5. Model evaluation - Evaluated the trained models on a separate test dataset to measure their performance in terms of accuracy and other relevant metrics.

  6. Comparative analysis - Compared the accuracy and results of each model to determine the best-performing model for the task of classifying images into children and adults categories.

Models Implemented

The following models were used in this project:

  1. VGG19 - VGG19 is a deep convolutional neural network known for its simplicity and effectiveness in image classification tasks. It consists of 19 layers and has achieved remarkable accuracy in various competitions and benchmarks.

  2. ResNet50 - ResNet50 is part of the ResNet (Residual Network) architecture, featuring residual connections that enable the training of very deep networks. ResNet50 specifically has 50 layers and has demonstrated superior performance in image classification tasks, especially on datasets with a large number of classes.

  3. InceptionV3 - InceptionV3, developed by Google, is famous for its inception module, which allows for efficient use of computational resources by parallelizing operations. It has been widely adopted due to its excellent trade-off between computational efficiency and accuracy,

  4. MobileNetV2 - MobileNetV2 is designed specifically for mobile and embedded vision applications, where computational resources are limited. It utilizes depth-wise separable convolutions to reduce the number of parameters and computations while maintaining high accuracy.

The choice of these models was based on their proven performance in image classification tasks and their varying architectural complexities, enabling a comprehensive analysis.

Conclusion

Based on the accuracy results, MobileNetV2 achieved the highest accuracy of 79% on the test dataset, making it the best-fitted model for this particular project. The other models also performed well but had slightly lower accuracies. This project demonstrates the effectiveness of deep learning models in classifying images of children and adults based on facial features, with potential applications in age estimation or child/adult recognition systems.

github-actions[bot] commented 2 hours ago

Thanks for creating the issue in ML-Nexus!πŸŽ‰ Before you start working on your PR, please make sure to:

github-actions[bot] commented 2 hours ago

Thanks for raising this issue! However, we believe a similar issue already exists. Kindly go through all the open issues and ask to be assigned to that issue.

UppuluriKalyani commented 27 minutes ago

@pavitraag proceed!