GitHub - yqx7150/HoloDiffusion: HoloDiffusion's code.

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
configs		configs
gt		gt
info		info
models		models
op		op
A_1k_PCsampling_demo.py		A_1k_PCsampling_demo.py
A_1k_arg_PCsampling_demo.py		A_1k_arg_PCsampling_demo.py
A_1k_arg_sampling.py		A_1k_arg_sampling.py
A_1k_arg_sampling_old.py		A_1k_arg_sampling_old.py
A_1k_sampling.py		A_1k_sampling.py
A_PCsampling_demo.py		A_PCsampling_demo.py
A_sampling.py		A_sampling.py
HoloDiffusion.ipynb		HoloDiffusion.ipynb
README		README
a.py		a.py
a2.py		a2.py
aapm_sin_ncsnpp_gb.py		aapm_sin_ncsnpp_gb.py
abcd_cut.mat		abcd_cut.mat
abcd_ori.mat		abcd_ori.mat
abcd_resize.mat		abcd_resize.mat
amp_gt_b.mat		amp_gt_b.mat
code		code
collect_csv.py		collect_csv.py
datasets.py		datasets.py
default_lsun_configs.py		default_lsun_configs.py
evaluation.py		evaluation.py
func_test.py		func_test.py
holo_everything.csv		holo_everything.csv
holo_tool.py		holo_tool.py
likelihood.py		likelihood.py
losses.py		losses.py
main.py		main.py
phase_gt_b.mat		phase_gt_b.mat
radon_utils.py		radon_utils.py
run_lib.py		run_lib.py
sampling_train.py		sampling_train.py
sde_lib.py		sde_lib.py
tt.py		tt.py
utils.py		utils.py

Repository files navigation

docker's environment:(cuda10.2,ubuntu16.04) 
docker pull zieghart/base:C10U16_perfect 
then:
conda activate ncsn

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

test size and gap:
python A_1k_arg_PCsampling_demo.py --size=500 --gpu=0 --gap=60 --useNet=True

test sensor number: 
python A_1k_arg_PCsampling_demo.py --size=500 --gpu=0 --gap=60 --ssn=2 --useNet=True 
python A_1k_arg_PCsampling_demo.py --size=500 --gpu=0 --gap=60 --ssn=3 --useNet=True

test sparse: 
python -u A_1k_arg_PCsampling_demo.py --size=500 --gpu=1 --gap=60 --sparse=6 --useNet=True # means SR = 5/6 
python -u A_1k_arg_PCsampling_demo.py --size=500 --gpu=1 --gap=60 --sparse=5 --useNet=True # means SR = 4/5

background_test(for long time test): 
nohup python -u A_1k_arg_PCsampling_demo.py --size=500 --gpu=0 --gap=60 > ./records/your_path/S500_G60.out 2>&1 &

if have any questions：
[email protected]