Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Images

Related tags

Deep Learning SASSnet
Overview

SASSnet

Code for paper: Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Images(MICCAI 2020)

Our code is origin from UA-MT

You can find paper in Arxiv.

Usage

  1. Clone the repo:
git clone https://github.com/kleinzcy/SASSnet.git 
cd SASSnet

  1. Put the data in data/2018LA_Seg_Training Set.

  2. Train the model

cd code
# for 16 label
python train_gan_sdfloss.py --gpu 0 --label 16 --consistency 0.01 --exp model_name
# for 8 label
python train_gan_sdfloss.py --gpu 0 --label 8 --consistency 0.015 --exp model_name

Params are the best setting in our experiment.

  1. Test the model
python test_LA.py --model model_name --gpu 0 --iter 6000

Our best model are saved in model dir.

Citation

If you find our work is useful for you, please cite us.

Comments
  • comparative experiment

    comparative experiment

    Dear author：

    Hi, Thanks for your great work in 《Shape-aware Semi-supervised 3D SemanticSegmentation for Medical Images》!

    This article has benefited me a lot.

    I am writing to you this time mainly to think about the code you are requesting for a comparative experiment.

    DAP、 ASDNet 、TCSE

    I can't find the relevant code on the Internet, so I'm very much looking forward to getting your comparative experiment code.

    Looking forward to hearing from you, I wish you all the best.

    opened by XYZera 2
  • about the test

    about the test

    Dear author：

    Hi, Thanks for your great work! I am interested in your work.

    Sorry to disturb you.I have some questions. I downloaded the "2018 Atrial Segmentation Challenge COMPLETE DATASET",and then used the "la_heart_processing.py" file to process the "Training Set", and put in in the corresponding directory，and the file “train_gan_sdfloss” work on, training success. But when I test with the train model "iter_6000.pth", the result showed :average metric is [ 0.18091923 0.10030632 53.18094769 23.59822443]. If I test with your "best.pth" it can get the same result as your paper showed.

    Can you give me some answers? Thanks

    Best wish!

    opened by 18677064404 2
  • about class number

    about class number

    Hi, thanks for your reply, I meet a question in the code, can you help, num_classes = 2, but during training, the input class number is n_classes=num_classes-1: net = VNet(n_channels=1, n_classes=num_classes-1, normalization='batchnorm', has_dropout=True) Can you tell me why do this operation? Many thanks, waiting for your reply.

    opened by GuobinZhangTJU 2
  • data

    data

    Your work is very good.I am very eager to download the data you use in your work.But the data can no longer be downloaded from the official website.Would you please share it with me?

    opened by ryandok 2
  • Question of your VNet result

    Question of your VNet result

    Dear author,

    I ran the baseline framework Vnet (highlights in picture below) based on the 8 label supervised protocol, and find that it is far behind the result on your table. image

    What I have is: 75.4014 61.832541 9.92698021 33.91709438. I'm wondering whether you have this supervised result from the Bayesian VNet, or from your model (i.e., with the assistance of the discriminator).

    kind regards,

    opened by ghost 1
  • Questions about the effect of SDF on 2D images

    Questions about the effect of SDF on 2D images

    Hello, I am very sorry to disturb you. First of all, thank you very much for your excellent work, and I have a question. Your experiments were done on 3D images, and they have excellent results. I would like to ask you if this SDF map will work on 2D images, and I feel that this distance map should also work on 2D images, right? I hope you can answer me, thank you very much!

    opened by woaicv 3
Owner
klein
2016-2020:Undergraduate student in WHU. 2020-now:Graduate student in Shanghai tech university, websit http://plus.sist.shanghaitech.edu.cn/
klein
GitHub
Semi-supervised Semantic Segmentation with Directional Context-aware Consistency (CVPR 2021)

Semi-supervised Semantic Segmentation with Directional Context-aware Consistency (CAC) Xin Lai*, Zhuotao Tian*, Li Jiang, Shu Liu, Hengshuang Zhao, Li

DV Lab 137 Dec 14, 2022
PyTorch implementation of ShapeConv: Shape-aware Convolutional Layer for RGB-D Indoor Semantic Segmentation.

Shape-aware Convolutional Layer (ShapeConv) PyTorch implementation of ShapeConv: Shape-aware Convolutional Layer for RGB-D Indoor Semantic Segmentatio

Hanchao Leng 82 Dec 29, 2022
Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.

Semi-supervised-learning-for-medical-image-segmentation. Recently, semi-supervised image segmentation has become a hot topic in medical image computin

Healthcare Intelligence Laboratory 1.3k Jan 3, 2023
The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

MIC-DKFZ 1.2k Jan 4, 2023
Anti-Adversarially Manipulated Attributions for Weakly and Semi-Supervised Semantic Segmentation (CVPR 2021)

Anti-Adversarially Manipulated Attributions for Weakly and Semi-Supervised Semantic Segmentation Input Image Initial CAM Successive Maps with adversar

Jungbeom Lee 110 Dec 7, 2022
[CVPR 2021] Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

TorchSemiSeg [CVPR 2021] Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision by Xiaokang Chen1, Yuhui Yuan2, Gang Zeng1, Jingdong Wang

Chen XiaoKang 387 Jan 8, 2023
ST++: Make Self-training Work Better for Semi-supervised Semantic Segmentation

ST++ This is the official PyTorch implementation of our paper: ST++: Make Self-training Work Better for Semi-supervised Semantic Segmentation. Lihe Ya

Lihe Yang 147 Jan 3, 2023
[ICCV 2021] A Simple Baseline for Semi-supervised Semantic Segmentation with Strong Data Augmentation

[ICCV 2021] A Simple Baseline for Semi-supervised Semantic Segmentation with Strong Data Augmentation

CodingMan 45 Dec 12, 2022
[CVPR 2022] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels

Using Unreliable Pseudo Labels Official PyTorch implementation of Semi-Supervised Semantic Segmentation Using Unreliable Pseudo Labels, CVPR 2022. Ple

Haochen Wang 268 Dec 24, 2022
[cvpr22] Perturbed and Strict Mean Teachers for Semi-supervised Semantic Segmentation

PS-MT [cvpr22] Perturbed and Strict Mean Teachers for Semi-supervised Semantic Segmentation by Yuyuan Liu, Yu Tian, Yuanhong Chen, Fengbei Liu, Vasile

Yuyuan Liu 132 Jan 3, 2023
Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation"

Medical-Transformer Pytorch Code for the paper "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" About this repo: This repo

Jeya Maria Jose 615 Dec 25, 2022
ISBI 2022: Cross-level Contrastive Learning and Consistency Constraint for Semi-supervised Medical Image.

Cross-level Contrastive Learning and Consistency Constraint for Semi-supervised Medical Image Introduction This repository contains the PyTorch implem

null 25 Nov 9, 2022
Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentation, CVPR 2018

Learning Pixel-level Semantic Affinity with Image-level Supervision This code is deprecated. Please see https://github.com/jiwoon-ahn/irn instead. Int

Jiwoon Ahn 337 Dec 15, 2022
An image base contains 490 images for learning (400 cars and 90 boats), and another 21 images for testingAn image base contains 490 images for learning (400 cars and 90 boats), and another 21 images for testing

SVM Données Une base d’images contient 490 images pour l’apprentissage (400 voitures et 90 bateaux), et encore 21 images pour fait des tests. Prétrait

Achraf Rahouti 3 Nov 30, 2021
Recall Loss for Semantic Segmentation (This repo implements the paper: Recall Loss for Semantic Segmentation)

Recall Loss for Semantic Segmentation (This repo implements the paper: Recall Loss for Semantic Segmentation) Download Synthia dataset The model uses

null 32 Sep 21, 2022
Official PyTorch implementation for paper Context Matters: Graph-based Self-supervised Representation Learning for Medical Images

Context Matters: Graph-based Self-supervised Representation Learning for Medical Images Official PyTorch implementation for paper Context Matters: Gra

null 49 Nov 23, 2022
Official and maintained implementation of the paper "OSS-Net: Memory Efficient High Resolution Semantic Segmentation of 3D Medical Data" [BMVC 2021].

OSS-Net: Memory Efficient High Resolution Semantic Segmentation of 3D Medical Data Christoph Reich, Tim Prangemeier, Özdemir Cetin & Heinz Koeppl | Pr

Christoph Reich 23 Sep 21, 2022
Build a medical knowledge graph based on Unified Language Medical System (UMLS)

UMLS-Graph Build a medical knowledge graph based on Unified Language Medical System (UMLS) Requisite Install MySQL Server 5.6 and import UMLS data int

Donghua Chen 6 Dec 25, 2022
[MedIA2021]MIDeepSeg: Minimally Interactive Segmentation of Unseen Objects from Medical Images Using Deep Learning

MIDeepSeg: Minimally Interactive Segmentation of Unseen Objects from Medical Images Using Deep Learning [MedIA or Arxiv] and [Demo] This repository pr

Healthcare Intelligence Laboratory 92 Dec 8, 2022