Error in predict_shapes: list index out of range

CVHub520 / X-AnyLabeling

Effortless data labeling with AI support from Segment Anything and other awesome models.

GNU General Public License v3.0

3.82k stars 442 forks source link

Error in predict_shapes: list index out of range #402

Closed hello50505 closed 3 months ago

hello50505 commented 4 months ago

1.我使用这个命令得到我的onnx文件： python3 export_yoloV8.py -w best.pt -s 1280 960 --dynamic （best.pt是我自己训练得到的yolov8s模型，class=1）

2.我在x-anylabeling使用自动标注，配置文件如下： type: yolov8 name: yolov8s-r20230520 display_name: YOLOv8s Ultralytics model_path: /home/xpc/Downloads/video_20240508_144952/best.onnx nms_threshold: 0.45 confidence_threshold: 0.25 input_width: 1280 input_height: 960 classes:

dingzi

3.当我点击推理图片的按钮后，命令窗口报错： Error in predict_shapes: list index out of range

请问问题出在哪里？谢谢！

CVHub520 commented 4 months ago

您好，暂不支持动态batch，可注释掉--dynamic参数重新导出。

hello50505 commented 4 months ago

您好，暂不支持动态batch，可注释掉--dynamic参数重新导出。

我把导出的命令改为： python3 export_yoloV8.py -w best.pt -s 1280 960 或者python3 export_yoloV8.py -w best.pt -s 1280 960 --batch 1

能正常导出onnx文件，但是自动标注还是会报同样的错误

CVHub520 commented 4 months ago

您好，暂不支持动态batch，可注释掉--dynamic参数重新导出。

我把导出的命令改为： python3 export_yoloV8.py -w best.pt -s 1280 960 或者python3 export_yoloV8.py -w best.pt -s 1280 960 --batch 1

能正常导出onnx文件，但是自动标注还是会报同样的错误

使用netron工具检查下你的onnx模型中classes类别是否与yaml文件的类别定义完全一致。

hello50505 commented 4 months ago

您好，暂不支持动态batch，可注释掉--dynamic参数重新导出。

我把导出的命令改为： python3 export_yoloV8.py -w best.pt -s 1280 960 或者python3 export_yoloV8.py -w best.pt -s 1280 960 --batch 1 能正常导出onnx文件，但是自动标注还是会报同样的错误

使用netron工具检查下你的onnx模型中classes类别是否与yaml文件的类别定义完全一致。

Screenshot from 2024-05-09 19-48-36 我自己看好像是没什么问题，class=1

CVHub520 commented 4 months ago

你这个输出节点没对应上，建议使用官方的导出设置。如果你仍然需要使用自定义的导出模型，可自行修改后处理代码进行适配。

hello50505 commented 4 months ago

你这个输出节点没对应上，建议使用官方的导出设置。如果你仍然需要使用自定义的导出模型，可自行修改后处理代码进行适配。

好的谢谢!我按照yolov8官方方法导出，目前可以自动标注了