config configs/culane.py

这是一个CULane数据集的配置文件，其中包含了训练、验证和测试的相关参数设置。

_base_ = [
    '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/culane.py',
    '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
]

# model settings
model = dict(
    decode_head=dict(num_classes=5), auxiliary_head=dict(num_classes=5))
    
# dataset settings
data = dict(
    samples_per_gpu=4,
    workers_per_gpu=4,
    train=dict(type='CULane', split='train'),
    val=dict(type='CULane', split='val'),
    test=dict(type='CULane', split='test'))

# optimizer
optimizer = dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=0.0001)
optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))

# learning policy
lr_config = dict(policy='poly', power=0.9, by_epoch=False)

# runtime settings
runner = dict(type='EpochBasedRunner', max_epochs=160)
checkpoint_config = dict(by_epoch=False, interval=20)
evaluation = dict(interval=20, metric='mIoU')

其中：

- `_base_` 表示使用的基础配置文件，这里使用了模型、数据集、运行时设置和学习率调度等基础配置文件。
- `model` 表示模型相关的设置，这里使用 FCN-R50-d8 作为基础模型，decode_head 和 auxiliary_head 都设置为 5 类别（即 5 条车道线）。
- `data` 表示数据集相关的设置，包括每个 GPU 用来训练的样本数、数据集划分方式等。
- `optimizer` 和 `optimizer_config` 表示优化器相关的设置，这里使用 SGD 优化器，设置了学习率、动量和权重衰减等参数。
- `lr_config` 表示学习率调度的设置，这里使用了 Poly 调度，设置了幂次和是否按 epoch 计算。
- `runner` 表示训练器相关的设置，这里使用了 EpochBasedRunner，并且设置了最大训练轮数。
- `checkpoint_config` 表示保存模型参数的设置，这里设置了每 20 轮保存一次模型，并且不按 epoch 计算。
- `evaluation` 表示验证时的评估设置，这里设置了每 20 轮进行一次验证，并且使用 mIoU 作为评估指标。