自动驾驶之 道路分割 使用deeplabv3模型
正在调参和训练中。。。请持续关注
作者:刘云飞
建议与合作联系邮箱:liuyunfei.1314@163.com
0x00 语言和工具
语言:Python 3.7
框架:PyTorch 1.2
网络结构:deeplab v3+ , 论文见paper文件夹
0x01 数据集
数据集采用kitti 数据集,分割所用的数据集比较费时费力,作者不再自行标注。
有精力自己标注数据集可以使用labelme标注。
Kitti datasets : http://www.cvlibs.net/datasets/kitti/
0x02 训练
由于kitti数据集只有200张,有点少,采用了mobilenet作为backbone进行训练。
PS:为啥不用ResNet等作为backbone呢?因为本人的GPU只有6G显存,只能跑起来mobilenet~~
有条件的可以用ResNet作为backbone,效果会更好。
0x03 结果
效果如下图,图片大小缩放到了512*512
视频已经上传到B站,链接:https://www.bilibili.com/video/av78626868/