画像セグメンテーションのコードをPyTorchで書き直してみました

自分の専門はテキストマイニングなのですが、最近は深層学習を利用した画像解析が普及しているので、勉強しておかなければと思い、下記のリンク先を読んでみました。

https://japan-medical-ai.github.io/medical-ai-course-materials/notebooks/Image_Segmentation.html

ここにあったMRI画像セグメンテーションのコードが非常に参考になったので、元はChainerで書かれていたものを、自分が普段PyTorchを使っていることもあり、PyTorchで書き直してみました。

https://github.com/tomonari-masada/medical-image-segmentation

訓練済みのモデル(model_epoch_1000.pth)と、そのモデルでvalidationデータの一枚について予測をさせた結果(temp.png)も置いてあります。このコードは、PyTorchのsuperresolutionのコードをフォークして書きました。ですので、元のChainerのコードとは、モデルを書いてある部分以外は別ものになっていると思います。

タグ: ,