如何使用unet网络进行训练和预测
环境配置
git clone https://github.com/zhixuhao/unet.git
D:
cd D:\KerasProject\unet
conda create -n unet python=3.6
activate unet
conda install tensorflow-gpu==1.12
conda install keras
pip install jupyter
pip install scikit-image
开始训练-使用pycharm
S1
Project地址是:
D:\KerasProject\unet
S2
使用pycharm打开
S3
修改路径
data/cta/test
修改成:
("data/CTANew/test")
注意,测试集里边的文件名要修改成
同时,还要把测试记得大小给修改成13张图片;
results = model.predict_generator(testGene,13,verbose=1)
S4
开始训练:
配置文件如下:
开始训练-使用jupyter-notebook
D:
cd D:\KerasProject\unet
activate unet
ipython notebook