Giter Club home page Giter Club logo

lesion-gan's Introduction

lesion-GAN

Code for "Towards Interpretable Skin Lesion Classification with Deep Learning Models"

HAM10000 Dataset

The dataset used for this project was the HAM10000 dataset, accessible at https://dataverse.harvard.edu/dataset.xhtml?persistentId=doi:10.7910/DVN/DBW86T

For our purposes, we used the 3 most prominent skin lesion classes.

Workflow

The "ISIC2018..." and "Processed Images" folders are placeholders for the dataset. After downloading the dataset, it should be placed in the first folder, and processed into the second. Throughout the code, tweaks will have to be made to the path and filenames to run on your machine.

The bulk of the project is in the numbered files, from 1 through 8, where the files should be ran in order. While running the files to convert the images into an numpy-compressed file, please pay attention to filenames.

The exception is the missing "7" file, which represents the CNN in the cnn_notebooks folder. While training the CNN (preferrably with a GPU), the balancedAccuracy.py file is needed to track accuracy during training, though it is not exactly accurate when training and averaging minibatches.

Additionally, after training the CNN (file "7") or multiple CNNs, the predictions of the CNNs on the data can be saved while it is being run on a virtual GPU through the getPredictions.ipynb file in the misc_scripts folder.

Important Variable References in code

"imageList" refers to the training set (roughly 64% of the modified dataset)

"imageValList" refers to the validation set (80% of the modified dataset in conjunction with imageList)

"testList" refers to the set-aside, testing set (remaining 20% of the modified dataset)

lesion-gan's People

Contributors

alxiang avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

Forkers

kukrak

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.