如何使用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

 

posted @ 2019-06-13 11:34  bH1pJ  阅读(404)  评论(0编辑  收藏  举报