从Github上克隆下来的项目来自https://github.com/zhixuhao/unet
1.首先克隆完项目,根据README,直接运行main.py就行(所需环境都配置好的情况下)。
2.报错如下:
1)model.py中有报input的错。
解决:将input,output改为inputs,outputs
2)name"Adam" not defined
解决:需要导入一个Adam的包,代码如下:
from keras.optimizer_v2.adam import Adam
3.运行
解决完上面两个问题以后就可以运行了,运行时间有点久,我电脑只能cpu运行,第一次运行了一个多小时。运行完以后会保存unet_membrane.hdf5文件,在data/membrane/test文件夹下会保存预测结果的图片。(源码里已经有训练好预测结果的图片,所以我把它们删了再运行的)
第一次运行结果不太好,loss我记得是0.6多,准确率0.7多。保存下来的图片如下图所示。
第二次运行完,结果就好多了