pip install tensorflow==1.14 keras tqdm opencv-python pillow libtiff
二、目录说明 data/building 数据根目录
models 网络定义
data.py 数据生成器
dice_loss.py 损失函数定义
test.py 测试
train.py 训练
如下图所示:
models 文件夹:
–data_path 数据根目录
– train 数据根目录下的训练图像文件夹
–label 数据根目录下的训练标签文件夹
–test 数据根目录下的测试图像文件夹(可以是None)
–test_label 数据根目录下的测试标签文件夹(可以是None)
–size 网络输入大小
–batch_size 训练batchsize
–epoch 训练总轮数
–resume 是否恢
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)