Skip to content

Repository for our paper "MFSNet: A Multi Focus Segmentation Network for Skin Lesion Segmentation"

License

Notifications You must be signed in to change notification settings

hritam-98/MFSNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MFSNet

This repository contains the official implementation of our paper titled "MFSNet: A Multi Focus Segmentation Network for Skin Lesion Segmentation" under peer review in Pattern Recognition, Elsevier.

Preprocessing

To run the script for inpainting, run the following using the command prompt:

python inpaint.py --root "D:/inputs/" --destination "D:/images/"

Training the Network

Follow the directory structure as follows:

+-- data
|   +-- .
|   +-- train
|   |   +-- images
|   |   +-- masks
|   +-- test
|   |   +-- images
|   |   +-- masks
+-- train.py
+-- test.py

Run the following to train the MFSNet network:

python train.py --train_path "data/train"

Other available hyperparameters for training are as follows:

  • --epoch: Number of epochs of training. Default = 100
  • --lr: Learning Rate. Default = 1e-4
  • --batchsize: Batch Size. Default = 20
  • --trainsize: Size of Training images (to be resized). Default = 352
  • --clip: Gradient Clipping Margin. Default = 0.5
  • --decay_rate: Learning rate decay. Default = 0.05
  • --decay_epoch: Number of epochs after which Learning Rate needs to decay. Default = 25

After the training is complete, run the following to generate the predictions on the test images:

python test.py --test_path "data/test"

Evaluating Performance

Run eval/main.m using MATLAB on the ground truth images and the predicted masks, to get the evaluation measures.

About

Repository for our paper "MFSNet: A Multi Focus Segmentation Network for Skin Lesion Segmentation"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published