Codes for "Super-resolved Virtual Staining of Label-free Tissue Using Diffusion Models" [paper]

Installation

Set up the environment using the conda environment file:

conda env create -f environment.yml

Train the model

python train.py

Download Test Samples and Pretrained Model

Download the test samples and pretrained model from the following Google Drive link. .

After downloading, place the "test_samples" and "models" folders under the same directory as the project.

Inference on Test Samples

To run inference using different sampling strategies, execute the following commands:

python test_diffusion_reverse_skip_sample_strategy.py
python test_diffusion_reverse_mean_sample_strategy.py

Evaluate results

To calculate the metrics, execute the following commands:

python analysis_metrics.py