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 details)

Uploaded Source

Built Distribution

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

Uploaded Python 3

File details

Details for the file ChexpertClassifier-0.0.1.tar.gz.

File metadata

  • Download URL: ChexpertClassifier-0.0.1.tar.gz
  • Upload date:
  • Size: 7.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.1 CPython/3.7.13

File hashes

Hashes for ChexpertClassifier-0.0.1.tar.gz
Algorithm Hash digest
SHA256 27bf6b4f64e15f74b78a90fe1a699466c27efd48a2a1d99358f2caf48cb48d8c
MD5 c72be4514ecdaa90e4facbb516551503
BLAKE2b-256 5d89ece0a1681917dab251803d97cbd7899b342a95d748315fd63281b7a83c19

See more details on using hashes here.

File details

Details for the file ChexpertClassifier-0.0.1-py3-none-any.whl.

File metadata

File hashes

Hashes for ChexpertClassifier-0.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 fea823bee8831761820edd2b84d48875d710ddee6dd33146484077669c2159c5
MD5 0690043754074df80a1adfae285afd4a
BLAKE2b-256 089127d748025b6235e2a7aebc54ea0aac1dab50b7b143f214e18759bd4260ee

See more details on using hashes here.

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