Git Product home page Git Product logo

blood-cell-classification's Introduction

Blood Cell Classification using MobileNetV2

This project uses transfer learning with MobileNetV2 to classify blood cell images into different types like red blood cells, white blood cells etc.

Dataset

The dataset used is from Kaggle ([https://www.kaggle.com/datasets/paultimothymooney/blood-cells]) It contains 12,500 augmented images of blood cells across 4 classes:

  • Eosinophil
  • Lymphocyte
  • Monocyte
  • Neutrophil

Requirements

Key libaries used:

  • TensorFlow 2.x
  • Matplotlib
  • scikit-learn
  • CUDA (Nvidia Graphics Required)

Preprocessing

The images are resized to 224x224 and pixel values are scaled between -1 and 1 before passing to MobileNetV2 model as the MobileNetV2 model was trained on that specific data.

Model

A pretrained MobileNetV2 model is used and the top classification layer is replaced for our 4 blood cell type classes. The model is then trained on the training images and evaluated on test set.

Training

The model is trained using categorical cross entropy loss and Adam optimizer. Callbacks are used for early stopping, model checkpointing and reducing learning rate on plateau. The images are split into training and test sets with training having validation split as well.

Evaluation

Accuracy of 94% was seen on the validation set.

Upcoming...

  • Use data from National Library of Medicine
  • Ill try newer architectures like EfficientNets
  • Experiment with somed hyperparameter tuning

blood-cell-classification's People

Contributors

prxshetty avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.