Skip to content

The Cardiomegaly Classification system is designed to classify chest X-ray images into categories related to the presence or absence of cardiomegaly.

Notifications You must be signed in to change notification settings

pranav-bot/Cardiomeology-Classification

Repository files navigation

Cardiomegaly Classification

Overview

The Cardiomegaly Classification system is designed to classify chest X-ray images into categories related to the presence or absence of cardiomegaly. This documentation provides information on the dataset, model architecture, training process, and evaluation metrics.

Dataset

NIH CXR8 Dataset

The NIH CXR8 dataset is a collection of chest X-ray images curated by the National Institutes of Health (NIH). It consists of images labeled with various thoracic pathology categories, including cardiomegaly. The dataset is widely used for training and evaluating deep learning models for medical image analysis.

Dataset Link: NIH CXR8 Dataset

Model Architecture

InceptionV3-Based Convolutional Neural Network (CNN)

The Cardiomegaly Classification model is built upon the InceptionV3 architecture, a deep convolutional neural network that has proven effective for image classification tasks. InceptionV3 is known for its ability to capture complex patterns and features in images through the use of inception modules.

Model Layers:

  • Input Layer: Accepts chest X-ray images of fixed dimensions.
  • Inception Blocks: Multiple blocks consisting of parallel convolutional and pooling operations with different filter sizes.
  • Global Average Pooling Layer: Reduces spatial dimensions and retains important features globally.
  • Fully Connected Layers: Make predictions based on features extracted by the InceptionV3 backbone.
  • Output Layer: Produces probability scores for the presence or absence of cardiomegaly.
  1. Data Preprocessing: Resizing, normalization, and augmentation of input images.
  2. Model Compilation: loss='binary_crossentropy', optimizer='adam'
  3. Training: Iteratively updating model weights using labeled training data.
  4. Validation: Assessing model performance on a separate validation set to avoid overfitting.

About

The Cardiomegaly Classification system is designed to classify chest X-ray images into categories related to the presence or absence of cardiomegaly.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published