Skip to content

Ultrasound Enhancement through Adversarial Generation - Workshop on Simulation and Synthesis in Medical Imaging - SASHIMI 2020

Notifications You must be signed in to change notification settings

BCV-Uniandes/UltraGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

UltraGAN: Ultrasound Enhancement Through Adversarial Generation

This repository provides a PyTorch implementation of UltraGAN, presented in the paper UltraGAN: Ultrasound Enhancement Through Adversarial Generation. Presented at the Simulation and Synthesis in Medical Imaging workshop SASHIMI in MICCAI, 2020. UltraGAN is a novel method for ultrasound enhancement that transfers quality details while preserving structural information. UltraGAN incorporates frequency loss functions and an anatomical coherence constraint to perform quality enhancement. We show improvement in image quality without sacrificing anatomical consistency.

Paper

UltraGAN: Ultrasound Enhancement Through Adversarial Generation
María Escobar 1* , Angela Castillo1*, Andrés Romero 2, Pablo Arbeláez1
*Equal contribution.
1 Center for Research and Formation in Artificial Intelligence (CINFONIA) , Universidad de Los Andes.
2 Computer Vision Lab, ETHZ

Usage

Cloning the repository

$ git clone https://github.com/BCV-Uniandes/UltraGAN.git

Train setup:

bash train_net.sh

Test setup:

bash test_net.sh

Citation

@inproceedings{escobar2020ultragan,
  title={UltraGAN: Ultrasound Enhancement Through Adversarial Generation},
  author={Escobar, Maria and Castillo, Angela and Romero, Andr{\'e}s and Arbel{\'a}ez, Pablo},
  booktitle={International Workshop on Simulation and Synthesis in Medical Imaging},
  pages={120--130},
  year={2020},
  organization={Springer}
}

Acknowledgments

Our code borrows heavily from CycleGAN, we thank the authors for their contributions to the community. The CAMUS dataset used in training and evaluation of our method can be found here.

About

Ultrasound Enhancement through Adversarial Generation - Workshop on Simulation and Synthesis in Medical Imaging - SASHIMI 2020

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published