Giter Club home page Giter Club logo

skin-disease-detection's Introduction

18CSC305J - AI

Domain : Machine Learning / AI

Problem Statement

Build Mathematical model using ML and signal processing concept to process the skin issues into dryness and oiliness.

Description

Patients face a lot of problems with their skin. The main issue remains with the type of skin they have. Depending on the type of skin various skin diseases can occur. So if the skin is too dry it may start to peel and lead to various skin diseases like Psoriasis, Eczema etc. likewise too oily skin can lead to pimples which in turn can lead to scars. So the identification of skin type is of utmost need.

Classes of Skin Diseases

alt text

Frontend

alt text

alt text

alt text

alt text

alt text

alt text

Tech Stacks

  • Pytorch
  • torchvision
  • Pillow
  • HTML
  • CSS
  • Bootstrap
  • Javascript
  • Flask
  • Werkzeug

Model Used

EfficientNet-B0

The EfficientNet-B0 architecture wasn’t developed by engineers but by the neural network itself. This model was developed using a multi-objective neural architecture search that optimizes both accuracy and floating-point operations.

Taking B0 as a baseline model, the authors developed a full family of EfficientNets from B1 to B7 which achieved state of the art accuracy on ImageNet while being very efficient to its competitors. Below is a table showing the performance of EfficientNets family on ImageNet dataset.

alt text

The Efficient-B0 is mobile sized architecture having 11M trainable parameters. This is what the architecture looks like:

alt text

The pre-trained imagenet weights were loaded into the model. The first seven layers of the model architecture were frozen and the rest of the layers were trained on the data. The Stochastic Gradient Descent optimizer was used with a learning rate of 0.01 and a momentum of 0.9 . The loss function used was categorical cross entropy. The model achieved an accuracy of 95.5% on the test data set.

Running Instructions

Open the terminal and type the following

$ git clone https://github.com/MahimaKhatri/Skin-Care.git
$ cd Skin-Care
$ python3 -m venv skin-class-env
$ source skin-class-env/bin/activate
$ pip3 install -r requirements.txt
$ python3 predict.py -m "Path to torch model" -i "Path to image"

Developers

Mudit Jindal

Mudit Jindal

Mahima Khatri

Mahima Khatri

Charvy Gugalia

Charvy

skin-disease-detection's People

Contributors

mahimakhatri avatar mudit14224 avatar winee165 avatar

Stargazers

Parul Dwivedi avatar  avatar  avatar  avatar Anil Rupnar avatar  avatar Anshul Hriday Patel avatar  avatar Dr. Artificial曾小健 avatar Peter Wayne avatar  avatar Esteban Zelaya avatar Pujaningsih avatar Mattia Neroni 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.