WaveCT-AIN_base_version

（Style Transfer for Medical Image）

Getting Started

Dependency

PyTorch >= 0.4.1
Check the requirements.txt

pip install -r requirements.txt

Arguments

--content: FOLDER-PATH-TO-CONTENT-IMAGES
--style: FOLDER-PATH-TO-STYLE-IMAGES
--output: FOLDER-PATH-TO-OUTPUT-IMAGES
--image_size: output image size
-a: all transfer
-d: decoder transfer
-e: encoder transfer
-s: skip transfer

Introduction

Pretrained models can be found in the ./model_checkpoints
Content and style images can be found in the./examples
./transfer.py is the transfer code

Test the model:

CUDA_VISIBLE_DEVICES=5 python transfer.py --content ./examples/content --style ./examples/style --output ./outputs/ --verbose --image_size 512 -d

The test results will be saved to ./outputs by default.

Result

Please cite:

Liu, Zhendong, et al. "Remove appearance shift for ultrasound image segmentation via fast and universal style transfer." 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI). IEEE, 2020.