YOLOP复现

代码下载

链接:https://pan.baidu.com/s/19FLVk_57Lrax0vkLUorKng
提取码:u8ed

环境准备

利用docker 快速建立本机上的环境

cd ${Project root}
docker build -t dl/training . 

docker 容器创建

docker run -it --gpus all -v /home/git/YOLOP:/home/YOLOP dl/training

数据准备

BDD100K的mini版本,并且按照官方repo里的要求进行排版

链接:https://pan.baidu.com/s/1r7FbMX2_OcDKsfrhujSfAQ
提取码:yc90

配置修改

在./lib/config/default.py中修改相关参数
修改第九行_C.WORKERS = 0
修改96行,121行_C.TRAIN.BATCH_SIZE_PER_GPU 为合适大小
根据注释修改53-56行路径(自定义数据集,包括影像和各类标签)

demo测试结果

python tools/demo.py --source inference/videos/1.mp4

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_42990464/article/details/126362567