paddle-PaddleSeg分割框架解读[02] 配置文件config详解

PaddleSeg分割框架解读[02] 配置文件config详解

以DeepLabv3+为例进行讲解

# 设定batch_size的值即为单卡迭代一次送入网络的图片数量，一般显卡显存越大，batch_size的值可以越大
batch_size: 2 
# epoch=(iters*batch_size*num_gpus)/N   (x*2*4)/28990=80
iters: 289900   # 289900/(28990/(2*4))=80个epochs  

# 训练数据设置
train_dataset:
  # 选择数据集格式
  type: Dataset
  # 选择数据集路径
  dataset_root: dataset/roadrisk