Kelasで3D U-Net (3DUnetCNN)を試してみようと思い、githubに示されているのデータセットと一緒にダウンロードした。
途中、ANTs
のN4BiasFieldCorrection
をダウンロードしてインストールしろとあるが、Anacondaのcondaを使って入れた。
ひとつめの躓き
指示の通りサブディレクトリの brats
で python
から
from preprocess import convert_brats_data
と打ったら
AttributeError: module 'keras.backend' has no attribute 'set_image_dim_ordering'
と言われてしまう。
ググると、どうやら最近のkeras.backend
にはset_image_dim_ordering
が無いらしく、代わりにkeras.backend.common
にその関数を見つけた。なので
unit3d/training.py
の12行目をK.set_image_dim_ordering('th')
→K.common.set_image_dim_ordering('th')
と書き換えたらうまくいった。
2つめの躓き
データをコンバートするために同じくpython上で
convert_brats_data("data/original", "data/preprocessed")
と唱えろとあるが、成果物ができるはずのdata/preprocessed
は空のままで、エラーも出ない。
おかしいと思い、コードを見てみた。
data/original
にデータを設置するためのフォルダの階層が一つ足りないだけでした。凡ミス。