Giter Club home page Giter Club logo

cancer_detection_cnn's Introduction

cancer_detection_cnn

CNN that can detect various types of skin cancer

Problem statement

To build a CNN based model which can accurately detect melanoma. Melanoma is a type of cancer that can be deadly if not detected early. It accounts for 75% of skin cancer deaths. A solution which can evaluate images and alert the dermatologists about the presence of melanoma has the potential to reduce a lot of manual effort needed in diagnosis.

You can download the dataset here

The dataset consists of 2357 images of malignant and benign oncological diseases, which were formed from the International Skin Imaging Collaboration (ISIC). All images were sorted according to the classification taken with ISIC, and all subsets were divided into the same number of images, with the exception of melanomas and moles, whose images are slightly dominant. The data set contains the following diseases:

  • Actinic keratosis
  • Basal cell carcinoma
  • Dermatofibroma
  • Melanoma
  • Nevus
  • Pigmented benign keratosis
  • Seborrheic keratosis
  • Squamous cell carcinoma
  • Vascular lesion

Project Pipeline

  • Data Reading/Data Understanding → Defining the path for train and test images
  • Dataset Creation→ Create train & validation dataset from the train directory with a batch size of 32. Also, make sure you resize your images to 180*180.
  • Dataset visualisation → Create a code to visualize one instance of all the nine classes present in the dataset
  • Model Building & training :
    • Create a CNN model, which can accurately detect 9 classes present in the dataset. While building the model rescale images to normalize pixel values between (0,1).
    • Choose an appropriate optimiser and loss function for model training
    • Train the model for ~20 epochs
    • Write your findings after the model fit, see if there is evidence of model overfit or underfit
  • Choose an appropriate data augmentation strategy to resolve underfitting/overfitting
  • Model Building & training on the augmented data :
    • Create a CNN model, which can accurately detect 9 classes present in the dataset. While building the model rescale images to normalize pixel values between (0,1).
    • Choose an appropriate optimiser and loss function for model training
    • Train the model for ~20 epochs
    • Write your findings after the model fit, see if the earlier issue is resolved or not?
  • Class distribution: Examine the current class distribution in the training dataset
    • Which class has the least number of samples?
    • Which classes dominate the data in terms of the proportionate number of samples?
  • Handling class imbalances: Rectify class imbalances present in the training dataset with Augmentor library.
  • Model Building & training on the rectified class imbalance data :
    • Create a CNN model, which can accurately detect 9 classes present in the dataset. While building the model rescale images to normalize pixel values between (0,1).
    • Choose an appropriate optimiser and loss function for model training
    • Train the model for ~30 epochs
    • Write your findings after the model fit, see if the issues are resolved or not?

cancer_detection_cnn's People

Contributors

gaurav-gilalkar 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.