该周主要是完成对于BiseNet的训练和测试以及模型的相关改进。
因为我们识别车道线,只需要将图片中的事物分为两类即可,第一类则是背景,第二类则是车道线。而同时有场地仍然没有确定,我只能拿自己的训练集进行相关的测试。
BiSeNetV2模型的参数相比于V1的参数更少,而模型更轻量级,速度更快,因此我选择了对于BiSeNet的训练修改,对于作者提出的网络,我进行了复现并训练,根据后面出现的问题再对网络进行一定的修正。
修改配置文件
## bisenetv2
cfg = dict(
model_type='bisenetv2',
n_cats=2,
num_aux_heads=4,
lr_start=