CS-Net: Channel and Spatial Attention Network for Curvilinear Structure Segmentation

This repo is the official implementation of CS-Net: Channel and Spatial Attention Network for Curvilinear Structure Segmentation.

The main contribution of this work is the publication of two scarce datasets in the medical image field. Plesae click the link below to access the details and source data.

DOWNLOAD

Experiment Results

Vessel Segmentation on Fundus

Vessel Segmentation on OCT-A images

Nerve fibre tracing on CCM

Note:

Requirements:

• PyTorch >= 0.4.1

• tqdm

• cv2

• visdom

• sklearn

The attention module was implemented based on DANet. The difference between the proposed module and the original block is that we added a new $1\times 3$ and $3 \times 1$ kernel convolution layer into spatial attention module. Plese refer to the paper for details.

Welcome Any Problems of This Project