Code Monkey home page Code Monkey logo

melanoma-detection-cnn-project's Introduction

Melanoma-Detection-Deep Leaning-Project

build a CNN-based model that detects melanoma. A data set will be provided, which contains images of malignant and benign tumours. Understand the problem statement carefully and go through the evaluation rubrics before solving the problem

Problem Statement :

build a multiclass classification model using a custom convolutional neural network in TensorFlow.

Problem statement: Build a CNN-based model that 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 that can evaluate images and alert dermatologists about the presence of melanoma can potentially reduce a lot of manual effort needed in diagnosis.

Documentation

You can download the data set from Here

The data set consists of 2,357 images of malignant and benign tumours, which were identified by the International Skin Imaging Collaboration (ISIC). All images were sorted according to the classification by ISIC, and all subsets were divided into the same number of images, except for melanomas and moles, whose images are slightly dominant.

The data set contains images relating to the following diseases:

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

Project Pipeline

Following are all the steps to consider while working on the data set and subsequently the model:

  • Data reading/data understanding: Define the path for training and testing images.
  • Data set creation: Create, train and validate the data set from the train directory with a batch size of 32. Also, make sure you resize your images to 180*180.
  • Data set visualisation: Create code to visualise one instance of all the nine classes present in the data set.
  • Model building and training: Create a model and report the findings. You can follow the below-mentioned steps:
    • Create a CNN model that can accurately detect nine classes present in the data set. While building the model, rescale images to normalise pixel values between (0,1).
    • Choose an appropriate optimiser and a loss function for model training.
    • Train the model for ~20 epochs.
    • Write your findings after the model fit. You must check if - there is any evidence of model overfit or underfit.
  • Resolving underfitting/overfitting issues: Choose an appropriate data augmentation strategy to resolve model underfitting/overfitting.
  • Model building and training on the augmented data: Follow the below-mentioned steps for building and training the model on augmented data:
    • Create a CNN model that can accurately detect nine classes present in the data set. While building the model, rescale images to normalise pixel values between (0,1).
    • Choose an appropriate optimiser and a loss function for model training.
    • Train the model for ~20 epochs.
    • Write your findings after the model fit, and check whether the earlier issue is resolved.
  • Class distribution: Examine the current class distribution in the training data set and explain the following:
    • 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 data set with the augmentor library.
  • Model building and training on the rectified class imbalance data: Follow the below-mentioned steps for building and training the model on the rectified class imbalance data:
    • Create a CNN model that can accurately detect nine classes present in the data set. While building the model, rescale images to normalise pixel values between (0,1).
    • Choose an appropriate optimiser and a loss function for model training.
    • Train the model for ~30 epochs.
    • Write your findings after the model fit, and check if the issues are resolved or not.

melanoma-detection-cnn-project's People

Contributors

dushyantnagar7806 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.