yolop 目标检测+车道线检测+可行驶区域检测
yolop目标检测+车道线检测+可行驶区域检测-深度学习文档类资源-CSDN下载
一、学习笔记
YOLOP: You Only Look Once for Panoptic Driving Perception是2021.8.25 在 cs.cv发表的一篇文章。YOLOP是一个用于多任务的端到端感知网,主要集中在交通目标检测,可驾驶区域分割和车道检测。
论文地址: https://arxiv.org/abs/2108.11250
二、测试
1、设备及环境
设备:jetson agx xavier
jetpsck:4.6.1
python3.6
虚拟环境(conda):torch1.10.0+torchvison0.11.1
2、环境搭建
Nvidia jetson系列 agx xavier/orin/nano/tx2 安装pytorch Gpu版_Ponnyao的博客-CSDN博客_xavier安装pytorch
3、下载源码
git clone https://github.com/hustvl/YOLOP
cd YOLOP
conda activate pytorch
pip3 install -r requirements.txt
4、测试
python tools/demo.py --source test.mp4 #视频
test.jpg #图像
0 #摄像头