yolov10自定义数据集上应用及遇到的问题
yolov10在定义数据集上应用及应用过程中遇到的问题。
·
- 下载安装
通过以下链接，下载yolov10：
https://github.com/THU-MIG/yolov10
文字可参考：https://arxiv.org/pdf/2405.14458
通过以下方式安装yolov10,(由于刚出来，还没能集成到ultralytics,官网上正在更新中)：
conda create -n yolov10 python=3.9
conda activate yolov10
pip install -r requirements.txt
pip install -e .
cd到yolov10存放目录，这是就可以用准备好的数据集进行训练和预测了。
- 遇到的问题
用predict预测时会提示如下错误：
yolo predict model=…\train2\weights\best.pt source=…\images imgsz=1024
File “f:\Code\yolov10-main\ultralytics\utils\ops.py”, line 216, in non_max_suppression
bs = prediction.shape[0] # batch size
AttributeError: ‘dict’ object has no attribute ‘shape’
修改ops.py，再出现的位置前面，添加prediction = prediction[‘one2many’][0]行，最终代码如下所示：
prediction = prediction[‘one2many’][0]
bs = prediction.shape[0]
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
11
0- 0
已为社区贡献2条内容
所有评论(0)