python 基于残差ResNeXt与UNet的优化网络

一、环境配置

   pip install tensorflow==1.14 keras tqdm opencv-python pillow libtiff

二、目录说明

   data/building 数据根目录
   models 网络定义
   data.py 数据生成器
   dice_loss.py 损失函数定义
   test.py 测试
   train.py 训练

   如下图所示：

   models 文件夹：

三、训练

3.1 模型训练

–data_path 数据根目录
– train 数据根目录下的训练图像文件夹
–label 数据根目录下的训练标签文件夹
–test 数据根目录下的测试图像文件夹（可以是None）
–test_label 数据根目录下的测试标签文件夹（可以是None）
–size 网络输入大小
–batch_size 训练batchsize
–epoch 训练总轮数
–resume 是否恢复训练

训练命令如下：
python train