一、环境配置
pip install tensorflow==1.14 keras tqdm opencv-python pillow libtiff
二、目录说明
data/building 数据根目录
models 网络定义
data.py 数据生成器
dice_loss.py 损失函数定义
test.py 测试
train.py 训练
三、代码
data.py
import os
import random
import numpy as np
import keras
from libtiff