YOLOP在不同数据集测试

参考code:https://github.com/hustvl/YOLOP

01测试

python tools/demo.py --source inference/images
测试图片:--source 图片路径 (或存放图片的文件夹路径)
测试视频:--source 视频路径 (或存放视频的文件夹路径)
相机实时:--source 0

 02demo

 修改第173行调用GPU

 03测试kitti_road数据

IndexError: boolean index did not match indexed array along dimension 0; dimension is 375 but corresponding boolean dimension is 194

解决:在demo.py的131行加入

img_det = cv2.resize(img_det, (da_seg_mask.shape[1], da_seg_mask.shape[0]), interpolation=cv2.INTER_AREA)

在plot.py的60行删掉

#   img = cv2.resize(img, (1280,720), interpolation=cv2.INTER_LINEAR)

测试result:

 04写在最后

别在秋招池子里泡着,

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
要训练自己的数据集,你可以按照以下步骤进行操作: 1. 下载数据集: 根据引用提供的链接和提取码,下载数据集并保存到本地的合适目录下。 2. 修改参数: 打开"./lib/config/default.py"文件,按照文档中的说明修改相关参数。这些参数包括数据集路径、模型参数等。 3. 开始训练: 使用以下命令开始训练模型: ``` python tools/train.py ``` 4. 模型推理: 在训练完成后,使用以下命令进行模型推理: ``` python tools/test.py --weights weights/End-to-end.pth ``` 请注意,以上步骤是基于YOLOP算法进行的。YOLOP是一个用于全景驾驶感知算法的工具,可以同时处理目标检测、可行驶区域分割和车道线检测这三个视觉任务,并在嵌入式设备上保持出色的实时性能和精度。你可以根据你的实际需求和数据集进行相应的调整和修改。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [YOLOP 训练+测试+模型评估](https://blog.csdn.net/Dora_blank/article/details/120070490)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [YOLOP个人数据代码复现](https://blog.csdn.net/small_wu/article/details/122712781)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值