Skip to content

Trained with just raw 3D volumes, a single Sli2Vol model can be used to propagate a single-slice annotation to the whole 3D volume, for any structures across different modalities.

Notifications You must be signed in to change notification settings

pakheiyeung/Sli2Vol

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Sli2Vol: Annotate a 3D Volume from a Single Slice with Self-Supervised Learning

Figure

This repository contains the codes (in PyTorch) for the framework introduced in the following paper:

Sli2Vol: Annotate a 3D Volume from a Single Slice with Self-Supervised Learning [Paper] [Project Page]

@article{yeung2021sli2vol,
	title = {Sli2Vol: Annotate a 3D Volume from a Single Slice with Self-Supervised Learning},
	author = {Yeung, Pak-Hei and Namburete, Ana IL and Xie, Weidi},
	booktitle = {International conference on Medical Image Computing and Computer Assisted Intervention},
	pages = {69--79},
	year = {2021},
}

Contents

  1. Dependencies
  2. Correspondence Flow Network
  3. Verification Module (TODO)

Dependencies

  • Python (3.6), other versions should also work
  • PyTorch (1.6), other versions should also work

Correspondence Flow Network

  1. The correspondence flow network as described in the paper is coded as the class Correspondence_Flow_Net in model.py
  2. It computes the affinity matrix between the input slice1_input and slice2_input and use the matrix to reconstruct slice2_reconstructed from the input slice1

About

Trained with just raw 3D volumes, a single Sli2Vol model can be used to propagate a single-slice annotation to the whole 3D volume, for any structures across different modalities.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages