Skip to main content

CheXpert Classification with EfficientNet B3

Project description

CheXpert Classification with EfficientNet B3 🫁⚕️

picture

This Package contains EfficientNet B3 model which has been trained over CheXpert Small version for 30 epochs and can be used to classify chest X-ray images for 14 classes including:

  • No Finding
  • Enlarged Cardiomediastinum
  • Cardiomegaly
  • Lung Opacity
  • Lung Lesion
  • Edema
  • Consolidation
  • Pneumonia
  • Atelectasis
  • Pneumothorax
  • Pleural Effusion
  • Pleural Other
  • Fracture
  • Support Devices

What is CheXpert?

CheXpert is a large dataset of chest X-rays this paper and competition for automated chest x-ray interpretation, which features uncertainty labels and radiologist-labeled reference standard evaluation sets.

picture

Why CheXpert? 🫁⚕️:

Chest radiography is the most common imaging examination globally, critical for screening, diagnosis, and management of many life threatening diseases. Automated chest radiograph interpretation at the level of practicing radiologists could provide substantial benefit in many medical settings, from improved workflow prioritization and clinical decision support to large-scale screening and global population health initiatives. For progress in both development and validation of automated algorithms, we realized there was a need for a labeled dataset that (1) was large, (2) had strong reference standards, and (3) provided expert human performance metrics for comparison.

Why is EfficientNet efficient? 🏛️:
  • EfficientNet uses a technique called compound coefficient to scale up models in a simple but effective manner. Instead of randomly scaling up width, depth or resolution, compound scaling uniformly scales each dimension with a certain fixed set of scaling coefficients.

  • EfficientNets have been the SOTA for high quality and quick image classification. They were released about 2 years ago and were quite popular for the way they scale which made their training much faster compared to other networks.

picture picture

Build Status Open In Colab

Prerequisites 🧰

  • Python 3.6
  • Pytorch
  • Numpy
  • Pandas
  • gdown

Accuracy 📈

Here is the Area Under ROC for the model:

picture

Features

  • Easy to use
  • Fast
  • Accurate

Usage

Pip install the package:

pip install ChexpertClassifier==0.0.1

Download Weights :

gdown https://drive.google.com/uc?id=1--QV0N-Zb2xJfQlhIoawREpT8sGH4zWA

Some Imports

from Chexpert_Classifier import chexpert_classifier

Run the model:

chexpert_classifier(pathInputImage = '/content/download (3).jpeg',pathOutputImage = 'heatmap_view1_frontal.png',pathModel = '/content/m-epoch0-12072022-085549.pth.tar')

Author

Name Github Home Page
Mehdi Hosseini Moghadam https://github.com/mehdihosseinimoghadam https://mehdihosseinimoghadam.github.io/

License

MIT

Free Software, Hell Yeah!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ChexpertClassifier-0.0.1.tar.gz (7.3 kB view hashes)

Uploaded Source

Built Distribution

ChexpertClassifier-0.0.1-py3-none-any.whl (7.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page