ogemarques/gan-skin-matlab

Example of how to use MATLAB to generate synthetic images of skin lesions.

Synthetic skin lesion image generation using GANs

This repository shows an example of how to use MATLAB to generate synthetic images of skin lesions.

The training data set consists of annotated images from the ISIC 2016 challenge, Task 3 (Lesion classification) data set, containing 900 dermoscopic lesion images in JPEG format¹.

The code is based on an example using the flowers dataset available online.

Snapshot of the GAN after training. (L) generated images; (R) generator (blue) and discriminator (red) curves showing score (between 0 and 1, where 0.5 is best) for each iteration.

Examples of synthetically generated images.

Requirements

• MATLAB 2021a or later
• Deep Learning Toolbox
• Parallel Computing Toolbox (only required for training using a GPU)

Suggested steps

1. Download or clone the repository.
2. Open MATLAB.
3. Edit the contents of the imageFolder variable in the skin_lesion_gan.mlx file to reflect the path to your selected dataset.
4. Run the skin_lesion_gan.mlx script and inspect results.

Additional remarks

• You are encouraged to expand and adapt the example to your needs.
• The choice of network architectures for both the generator and discriminator was inspired by the DCGAN.
• The choice of hyperparameters (learning rate, mini-batch size, number of epochs, etc.) is merely illustrative. You are encouraged to (use Experiment Manager to) tweak those choices and find a better solution ².

Notes

[1] This example uses a fairly small subset of images to make it easier to get started without having to worry about large downloads and long training times.

[2] See this for ideas and example code for using MATLAB Experiment Manager.