新人避坑(刚开始接触医学图像分割)
在刚开始跑网络分割模型的时候总是一个一个源码去跑,这就导致需要自己匹配代码的数据集格式,需要在图像预处理上花费较多时间。最好的方法是在源码的基础上将网络模型提取出来,在一个基本的图像分割网络进行调用,下面以U-net网络源码调用RegSeg为例。
https://github.com/RolandGao/RegSeg
2. 看U-net的网络结构文件和RegSeg的网络文件model.py
(在这之间需要注意的是RegSeg它的model.py在python里有同名的函数,所以在放入unet的时候需要改名比如RegSeg_model.py,以及看清楚model里面需要的其它模块需要一起复制过