Pytorch implementation of SwinCup: Cascaded Swin Transformer for Histopathological Structures Segmentation in Colorectal Cancer
Transformer models have recently become the dominant architecture in many computer vision tasks, including image classification, object detection, and image segmentation. The main reason behind their success is the ability to incorporate global context information into the learning process. By utilising self-attention, recent advancements in the Transformer architecture design enable models to consider long-range dependencies. In this paper, we propose a novel transformer, named Swin Transformer with Cascaded UPsampling (SwinCup) model for the segmentation of histopathology images. We use a hierarchical Swin Transformer with shifted windows as an encoder to extract global context features. The multi-scale feature extraction in a Swin transformer enables the model to attend to different areas in the image at different scales. A cascaded up-sampling decoder is used with an encoder to improve its feature aggregation. Experiments on GLAS and CRAG histopathology colorectal cancer datasets were used to validate the model.
If you find this helpful, please cite our paper:
- "Usama Zidan, Mohamed Medhat Gaber, Mohammed M. Abdelsamea,
SwinCup: Cascaded swin transformer for histopathological structures segmentation in colorectal cancer, Expert Systems with Applications, Volume 216, 2023."
Paper
pip install --process-dependency-links -e .