你确认删除该任务么?此任务一旦删除不可恢复。
huangweijian fab9d01fb4 | 2 年前 | |
---|---|---|
neuron | 2 年前 | |
pynd | 2 年前 | |
pytools | 2 年前 | |
src | 2 年前 | |
.gitignore | 2 年前 | |
Framework.jpg | 2 年前 | |
LICENSE | 2 年前 | |
README.md | 2 年前 | |
affine_train.py | 2 年前 | |
losses.py | 2 年前 | |
model.py | 2 年前 | |
plot.py | 2 年前 | |
registration.py | 2 年前 |
This is the code of TMI's paper 'A coarse-to-fine deformable transformation framework for unsupervised multi-contrast MR image registration with dual consistency constraint'.
Multi-contrast magnetic resonance (MR) image registration is useful in the clinic to achieve fast and accurate imaging-based disease diagnosis and treatment planning. Nevertheless, the efficiency and performance of the existing registration algorithms can still be improved. In this paper, we propose a novel unsupervised learning-based framework to achieve accurate and efficient multi-contrast MR image registrations. Specifically, an end-to-end coarse-to-fine network architecture consisting of affine and deformable transformations is designed to improve the robustness and achieve end-to-end registration. Furthermore, a dual consistency constraint and a new prior knowledge-based loss function are developed to enhance the registration performances. The proposed method has been evaluated on a clinical dataset containing 555 cases, and encouraging performances have been achieved. Compared to the commonly utilized registration methods, including VoxelMorph, SyN, and LT-Net, the proposed method achieves better registration performance with a Dice score of 0.8397 in identifying stroke lesions. With regards to the registration speed, our method is about 10 times faster than the most competitive method of SyN (Affine) when testing on a CPU. Moreover, we prove that our method can still perform well on more challenging tasks with lacking scanning information data, showing the high robustness for the clinical application.
Two steps for training are needed:
Python >= 3.6
CUDA == 11.0
Tensorflow-gpu == 1.10
Keras == 2.2.0
If you use this framework or some part of the code, please cite:
W. Huang et al., "A coarse-to-fine deformable transformation framework for unsupervised multi-contrast MR image registration with dual consistency constraint," in IEEE Transactions on Medical Imaging, doi: 10.1109/TMI.2021.3059282.
尊敬的启智用户
感谢您一直以来对Openl启智社区AI协作平台的支持。为了保障您的使用权益和确保网络安全,我们于2024年1月份更新了《Openl启智社区AI协作平台使用协议》。更新后的协议明确了用户禁止使用内网穿透工具的条例。您单击“同意并继续”后,便可以继续使用我们的服务。感谢您的合作与理解。
更多协议内容,请参考《Openl启智社区AI协作平台使用协议》