voxelmorph配置

简介

VoxelMorph使用CNN实现了非监督的医学图像配准,速度较之前的方法有很大提升。主要特点有:
提出了一种基于学习的解决方案,不需要在训练过程中获取诸如ground truth对应或解剖标志等信息;
提出一个参数跨种群共享的CNN函数,通过函数评估实现配准;
参数优化的方法可以使用各种代价函数,从而适应各种任务;

VoxelMorph这个组发表了多篇论文,有不同的使用场景,具体的区别请看论文,但代码都在这一个库

下载代码:

https://github.com/voxelmorph/voxelmorph
到本地后解压缩出来

配置环境:

本人环境:

ubuntu18.04 + Anaconda 4.7 + python3.7

然后安装tensorflow和keras,这里看我之前的文章 (这里voxelmorph用的是tensorflow1+)
所以如果你用的是tf2,可以根据下面那个官方教程,然后自己改
然后安装依赖

pip install nibabel
conda install -c conda-forge tqdm
pip install Pillow
pip install matplotlib

所有的安装完了之后,就可以运行程序了。

运行

先进到下载的代码的文件夹的src中,然后在终端上运行,:

python register.py   - -gpu  0    ../data/test_vol.nii.gz    ../data/atlas_norm.nii.gz     --out_img     ../data/out.nii.gz      –model_file       ../models/cvpr2018_vm2_cc.h5

运行后的结果在data文件件下的out.nii.gz中,如何查看,请看上篇文章

一个官方的教程:
https://www.kaggle.com/adalca/learn2reg
我根据这个教程,实现minist的时候,需要以下的文件,然后自己改下import就可以运行了

更新个一个更简洁的入门代码,在kaggle的基础上修改源码,达到用最少的代码跑起来例子的效果

https://github.com/FlyuZ/voxelmorph-tf2-min

posted @ 2019-10-23 20:13  feifei97  阅读(5714)  评论(3编辑  收藏  举报