CELL-Diff-VS

Diffusion based virtual staining with CELL-Diff

Installation

To set up CELL-Diff-VS, begin by creating a conda environment:

conda create --name cell_diff_vs python=3.10

Activate the environment and run the installation script:

conda activate cell_diff_vs
bash install.sh

Training

Note: Edit the run_*.sh scripts to set the appropriate model and data paths before running.

bash run_train.sh

Validation

Note: Edit the run_*.sh scripts to set the appropriate model and data paths before running.

bash run_eval.sh

Testing

Note: Edit the run_*.sh scripts to set the appropriate model and data paths before running.

For HEK293T Cell Dataset:

bash run_eval_hek.sh

For A549 Cell Dataset:

bash run_eval_a549.sh