Skip to content

[MediaEval Medico Challenge'2020]: Real-time polyp segmentation using U-Net with IoU loss

Notifications You must be signed in to change notification settings

GeorgeBatch/kvasir-seg

Repository files navigation

2020 MediaEval Medico Challenge: Polyp Segmentation

This repository accompanies our (George Batchkala and Sharib Ali) working-notes paper "Real-time polyp segmentation using U-Net with IoU loss" presented at MediaEval 2020 Multimedia Benchmark Workshop, which was held online on 14-15 December 2020. If you are interested in this work, we recommend you first getting familiar with the overview paper.

To sum up:

About the challenge

Disclaimer: next paragraph was directly copied from the official GitHub repository of the challenge: https://github.com/DebeshJha/2020-MediaEval-Medico-polyp-segmentation.

The “Medico automatic polyp segmentation task” aims to develop computer-aided diagnosis systems for automatic polyp segmentation to detect all types of polyps (for example, irregular polyp, smaller or flat polyps) with high efficiency and accuracy. The main goal of the challenge is to benchmark semantic segmentation algorithms on a publicly available dataset, emphasizing robustness, speed, and generalization.

For more Information consult next section (Information and Links).

Information and Links

Repository Contents

  • train_models.py:

    • script to train the models. Use these commands to reproduce the weights (you might need to adjust the batch size if your GPU memory is different).
    python train_models.py --loss_function="IoULoss" --training_augmentation=0
    python train_models.py --loss_function="BCEWithLogitsLoss" --training_augmentation=0
    python train_models.py --loss_function="IoUBCELoss" --training_augmentation=0
    python train_models.py --loss_function="IoULoss" --training_augmentation=1
    python train_models.py --loss_function="BCEWithLogitsLoss" --training_augmentation=1
    python train_models.py --loss_function="IoUBCELoss" --training_augmentation=1
  • checkpoints:

    • Checkpoints for model weights, best epoch indexes, and validation IoU. More information and the link for download in the README.md inside the folder.
  • data:

  • models:

    • Different architectures. Each in its own script.
  • notebooks:

    • All notebooks: EDA, validation results, testing predictions.
  • presentation:

    • Presentation slides for self-organized mini-conference at the Health Data Science CDT after completing the Deep Learning challenge.
  • submission

    • Notebook and script with submission code.
    • Functionality: loads images one by one, makes predictions and records mean number of images/frames processed per second (FPS).
  • train-val-split:

  • utils:

    • metrics.py: loss and evaluation metrics
    • dataset.py: custom Dataset class to be used in torch DataLoaders

Environment

These are merely for reference.

python=3.8.10
torch=1.10.2 # this is PyTorch (I had it with cudatoolkit 10.2)
torchvision=0.11.3
imageio=2.26.0

torchsummary=1.5.1
torchmetrics=0.9.1

About

[MediaEval Medico Challenge'2020]: Real-time polyp segmentation using U-Net with IoU loss

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published