Skip to content

harishanand95/cxr_classification

Repository files navigation

This is attempt at classifying CXR images into Tuberculosis and non-tuberculosis categories

[Under-development] JSRT lung nodule classification (and locating) over CT scan images. https://github.com/harishanand95/jsrt-parser

China Set - The Shenzhen set - Chest X-ray Database Description: The standard digital image database for Tuberculosis is created by the National Library of Medicine, Maryland, USA in collaboration with Shenzhen No.3 People’s Hospital, Guangdong Medical College, Shenzhen, China. The Chest X-rays are from out-patient clinics, and were captured as part of the daily routine using Philips DR Digital Diagnose systems. Number of X-rays:

  • 336 cases with manifestation of tuberculosis, and
  • 326 normal cases.

It is requested that publications resulting from the use of this data attribute the source (National Library of Medicine, National Institutes of Health, Bethesda, MD, USA and Shenzhen No.3 People’s Hospital, Guangdong Medical College, Shenzhen, China) and cite the following publications:

  • Jaeger S, Karargyris A, Candemir S, Folio L, Siegelman J, Callaghan F, Xue Z, Palaniappan K, Singh RK, Antani S, Thoma G, Wang YX, Lu PX, McDonald CJ. Automatic tuberculosis screening using chest radiographs. IEEE Trans Med Imaging. 2014 Feb;33(2):233-45. doi: 10.1109/TMI.2013.2284099. PMID: 24108713
  • Candemir S, Jaeger S, Palaniappan K, Musco JP, Singh RK, Xue Z, Karargyris A, Antani S, Thoma G, McDonald CJ. Lung segmentation in chest radiographs using anatomical atlases with nonrigid registration. IEEE Trans Med Imaging. 2014 Feb;33(2):577-90. doi: 10.1109/TMI.2013.2290491. PMID: 24239990

Here is the image of an attempted small model.

  • Input is in 640x480 image in greyscale.
  • Images are split across as 182 validation, 180 test and 300 train images.
  • (640x480x1) input after conv1 operation of stride 2 and with 16 filters changes to (320x240x16).
  • (320x240x16) input after conv2 operation of stride 2 and with 16 filters changes to (160x120x16).
  • Relu and L2 regularisation is used.
  • A fully connected layer follows it and flattens it to 64 neurons.
  • Dropout is used as normalisation after previous layer.
  • A final FC layer with softmax function outputs a score on 2 values (TB,NoTB).
  • Adam optimizer is used.

alt text

classification_cxr.py has a model made in tensorflow itself while tflearn_classify has tflearn library used.

classification_cxr.py 's model

alt text

About

Classification CXR images from NLM-ChinaCXRSet Dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages