yolov10自定义数据集上应用及遇到的问题

1. 下载安装
   通过以下链接，下载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存放目录，这是就可以用准备好的数据集进行训练和预测了。

1. 遇到的问题
   用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]