/CCBANet

CCBANet: Cascading Context and Balancing Attention for Polyp Segmentation

Primary LanguagePython

CCBANet

Introduction

This repository contains the PyTorch implementation of CCBANet,Cascading Context and Balancing Attention for Polyp Segmentation, MICCAI 2021.

Install dependencies

  • torch
  • torchvision
  • opencv
  • Albumentations

Usage

1. Training

python train.py --dataset "dataset-name" --batch_size batch-size --load_ckpt "/path-to-check-point" --epoch_start epoch-start

2. Inference

python test.py --dataset "dataset-name" --batch_size 1 --load_ckpt "/path-to-check-point"

Acknowledgement

Part of the code was adpated from ACSNet:Adaptive Context Selection for Polyp Segmentation