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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM