Thanks to visit codestin.com
Credit goes to github.com

Skip to content

ShunPengFR/AvmPept

Repository files navigation

AvmPept

环视相机全景图感知多任务,包括停车位、可行使区域、减速带、车道线、道路箭头检测。

(备注:开源模型,开源数据,开源框架,非量产、非工程方案)

1 依赖

data_loader, model用pytorch搭建,训练测试runner用mmengine搭建。

  • 创建conda环境,安装pytorch(其它版本应该也行):
    conda create -n avmpept python=3.8 pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.3 -c pytorch -y
  • mmengine安装:pip install -e . -v
  • labelme安装:pip install labelme

2 代码简介

1) tools文件夹:单任务和多任务的训练、测试和推理脚本,任务包含停车位检测(pld)、可行使区域检测(freespace)和多任务(multi_task)。

2) projects文件夹:包含数据处理(dataset)、模型(model)、损失函数(loss)和指标(metric)

3 方案简介(持续优化)

(非常感谢博登智能分享提供的环视泊车数据集,本人仅在学习中使用)

  • 数据增强:亮度调节、旋转、翻转
  • backbone+neck:resnet18, FPN, 尝试使用ASPP(语义分割deeplabv3)
  • pld_head:车位类型,角点检测,进入线,分割线检测,focal_loss, L1_loss
  • fs_head:语义分割,bce_loss, lovasz_loss

4 测试

测试数据:https://zhuanlan.zhihu.com/p/564718292

权重下载:待上传

视频效果:https://www.bilibili.com/video/BV1xYade6EK4/?spm_id_from=333.999.0.0&vd_source=2a07defc3887b243207c8c9e58827595

https://space.bilibili.com/328243224/video

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages