代码下载
链接: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