By Rafael, Flávio and Romuere.
The massive use of multimedia technologies has enabled the exploration of information in many data such as texts, audio, videos, and images. Computational methods are being developed for several purposes such as monitoring, security, business, and even health through the automatic diagnosis of diseases by medical images. Among these diseases, we have melanoma skin cancer. Melanoma is a skin cancer that causes a large number of fatalities worldwide. Several methods for the automatic diagnosis of melanoma in dermoscopic images have been developed. For these methods to be more efficient, it is essential to isolate the lesion region. This study used a melanoma segmentation method based on U-net and Linknet deep learning networks combined with transfer learning and fine-tuning techniques. Additionally, we evaluate the model's ability to learn to segment the disease or just the dataset by combining datasets. The experiments were carried out in three datasets (PH2, ISIC 2018, and DermIS) and obtained promising results, with emphasis on the U-net that obtained an average Dice of 0.923 in the PH2 dataset, Dice = 0.893 in ISIC 2018, and Dice = 0.879 in the DermIS dataset.
Create a virtual environment and install the dependencies
pip install -r requeriments.txt
Open the * .py file and set values for:
- Choose the GPU at:
GPU_GLOBAL
- Choose the Dataset (0 - ph2, 1 - dermis, 2 - isic2018) in:
base_escolhida
- Choose the folder where the tests will be saved (don't forget to create the folder before running the code):
PASTA_DE_TESTES
- Choose batch_size at:
BATCH_SIZE_GLOBAL
- Choose the number of seasons in:
NUMERO_EPOCAS_GLOBAL
Run the *.py file:
python *.py