GitHub - yqx7150/OSDM: One Sample Diffusion Model in Projection Domain for Low-Dose CT Imaging

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
Test_CT		Test_CT
Test_CT1		Test_CT1
__pycache__		__pycache__
configs		configs
models		models
op		op
tv		tv
B_sampling(1).py		B_sampling(1).py
B_sampling.py		B_sampling.py
C_aapm_sin_ncsnpp_gb.py		C_aapm_sin_ncsnpp_gb.py
C_default_lsun_configs.py		C_default_lsun_configs.py
PCsampling_demo.py		PCsampling_demo.py
README		README
datasets.py		datasets.py
evaluation.py		evaluation.py
func_test.py		func_test.py
ideal_img.mat		ideal_img.mat
ideal_sinogram.mat		ideal_sinogram.mat
imagetest.py		imagetest.py
img_ldct.mat		img_ldct.mat
info.rar		info.rar
likelihood.py		likelihood.py
losses.py		losses.py
main.py		main.py
radon_utils.py		radon_utils.py
rec_img.mat		rec_img.mat
rec_sinogram.mat		rec_sinogram.mat
run_lib.py		run_lib.py
sampling.py		sampling.py
sampling1e5.py		sampling1e5.py
sampling_3noise_fujian_mask.py		sampling_3noise_fujian_mask.py
sampling_train.py		sampling_train.py
sde_lib.py		sde_lib.py
sinogram_ldct.mat		sinogram_ldct.mat
tvdenoise.py		tvdenoise.py
utils.py		utils.py

Repository files navigation

train:
python main.py --config=aapm_sin_ncsnpp_gb.py --workdir=exp --mode=train --eval_folder=result

test:
python PCsampling_demo.py

test默认调用exp_demo下的模型

--workdir=exp_zl
--mode=train
--eval_folder=result
--config=aapm_sin_ncsnpp_gb.py

CUDA_VISIBLE_DEVICES=1 python main.py --config=aapm_sin_ncsnpp_gb.py --workdir=exp_zl --mode=train --eval_folder=result


vali[vali < 0] = 0