Unet实现图像分割(一)

大憨熊 提交于 2019-12-15 03:41:30

在Rrtina-Unet-master文件夹下新建一个test文件夹(源代码没有这个test文件夹会报错)
将lib文件夹下的三个py文件里面的from xxx import xxx格式改成一致,就是说from lib.help_function xxx 的这个lib要么都有要么都没有。不然会报错:No model named xxx。我的解决办法:将所有from后面的lib都删掉,然后在retinaNN_training.py里面加入下面箭头所指代码:
在这里插入图片描述
使用google的免费GPU进行训练,但是如果按源代码的参数的话,内存不足,所有改小参数。例如可以将Unet的卷积核参数改成原来的1/4,或者将配置文件里面的N_subimgs从190000改成38000之类的,或者吧预测图片的数量full_images_to_test改小。
将训练代码里面的100行和101行的可视化代码注释掉,111和112行也注释掉,很容易报错。

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!