DAR-UNet
DAR-UNet copied to clipboard
关于stage1的效果
您好,感谢你这项出色的工作,我有两个问题想要向您请教,
- 在stage_1.5_i2i_inference.py中,您给出了CT2MR的代码,我用您给出的数据中的 'source_training_npy'和'target_training_npy'文件夹下 的数据作为source和target进行了CT2MR的转换(使用权重'enc_40.pt').但是效果相较您转换好的('source2target_training_npy_multi_style'文件下的结果)有一定差距,想知道这是什么原因造成的?
- 为了进行MR2CT的转换,根据我的理解,我把stage_1.5_i2i_inference.py'中的
https://github.com/Kaiseem/DAR-UNet/blob/056d3fc2918d8a9056ad75f99c87ce45045406cd/stage_1.5_i2i_inference.py#L33
更改为:
s=trainer.enc_s_a(single_img).cpu().numpy()[0]
并交换了source和target的文件夹路径,不知道这样的做法是否正确?结果仍有差距,且差距相比问题1更大。
如图是CT2MR和MR2CT的转换结果: