Adaptive Context Selection for Polyp Segmentation

Introduction

This repository contains the PyTorch implementation of:

Adaptive Context Selection for Polyp Segmentation, MICCAI 2020.

Requirements

• torch
• torchvision
• tqdm
• opencv
• scipy
• skimage
• PIL
• numpy

Usage

1. Training

python train.py --mode train --dataset kvasir_SEG --train_data_dir /path-to-train_data --valid_data_dir /path-to-valid_data

2. Inference

python test.py --mode test --load_ckpt checkpoint --dataset kvasir_SEG --test_data_dir /path-to-test_data

Citation

If you feel this work is helpful, please cite our paper

@inproceedings{zhang2020adaptive,
title={Adaptive Context Selection for Polyp Segmentation},
author={Zhang, Ruifei and Li, Guanbin and Li, Zhen and Cui, Shuguang and Qian, Dahong and Yu, Yizhou},
booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
pages={253--262},
year={2020},
organization={Springer}
}