Official Repo of "MS-SST: Single Image Reconstruction-based Stain-Style Transfer for Multi-domain Hematoxylin & Eosin Stained Pathology Images"
by Juwon Kweon, Mujung Kim, Gilly Yun, Soonchul Kwon, and Jisang Yoo
This paper has been submitted for publication in IEEE Access.
.
You can download a copy of all the files in this repository by cloning the git repository:
git clone https://github.com/jwkweon/MS-SST.git
To install requirements:
pip install -r requirements.txt
To train the model(s) in the paper, run this command:
python train.py --input-data <path_to_data> --alpha 10 --beta 20
To evaluate my model on ImageNet, run:
python eval.py --model-file mymodel.pth --benchmark imagenet