/CPFNet_Project

This is a pytorch project about medical segmentation

Primary LanguagePythonMIT LicenseMIT

CPFNet_Project

Introduction

This repository is for the PAPER: CPFNet: Context Pyramid Fusion Network for Medical Image Segmentation, which has been accepted by IEEE TRANSACTIONS ON MEDICAL IMAGING

The author list: Shuanglang Feng, Heming Zhao, Fei Shi, Xuena Cheng, Meng Wang, Yuhui Ma, Dehui Xiang, Weifang Zhu and Xinjian Chen from SooChow University.

The source code is now available!

citation

@inproceedings{feng2020cpfnet,
    author={Shuanglang Feng and Heming Zhao and Fei Shi and Xuena Cheng and Meng Wang and Yuhui Ma and Dehui Xiang and Weifang Zhu and Xinjian Chen},
    title={CPFNet: Context Pyramid Fusion Network for Medical Image Segmentation},
    booktitle={IEEE TRANSACTIONS ON MEDICAL IMAGING},   
    year={2020},   
}

Folder

  • Dataset: the folder where dataset is placed.
  • OCT: the folder where model and model environment code are placed,OCT is the name of task.
    • dataset: the file of data preprocessing.
    • model: model files.
    • utils: utils files(include many utils)
      • config.py: some configuration about project parameters.
      • loss.py: some custom loss functions
      • utils.py: some definitions of evaluation indicators
    • metric.py: offline evaluation function
    • train.py: training, validation and test function.
  • Pretrain_model: pretriand encoder model,for example,resnet34.

Prerequisites

  • PyTorch 1.0
    • conda install torch torchvision
  • tqdm
    • conda install tqdm
  • imgaug
    • conda install six numpy scipy Pillow matplotlib scikit-image opencv-python imageio Shapely
    • conda install imgaug