MRI2CT_beta

Adapted from ControlNet for MRI to CT task.

Run

git clone https://github.com/lujiazho/MRI2CT_beta.git
cd MRI2CT_beta

mkdir data
cp /ifs/loni/faculty/shi/spectrum/Student_2020/lzhong/MRI2CT/data/image_pairs.json ./data

The Dataset will directly refer to the prepared data under my directory, so it's safe to just run

CUDA_VISIBLE_DEVICES=3 python train_MRI2CT.py

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
cldm		cldm
ldm		ldm
models		models
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
config.py		config.py
environment.yaml		environment.yaml
notes.md		notes.md
prepare_dataset.py		prepare_dataset.py
share.py		share.py
tool_add_control.py		tool_add_control.py
tool_add_control_sd21.py		tool_add_control_sd21.py
tool_transfer_control.py		tool_transfer_control.py
train_MRI2CT.py		train_MRI2CT.py