[Bug]无法使用自己训练的模型用于自动标注

[MichaelVi1866]

我发现只有作者你提供的yolov11n-seg.onnx能用来做自动标注，我自己官网上下载的yolov11n.pt转化为onnx后，我看了里面的模型构造，和作者你提供的onnx模型比，明显不一样。然后我在netron上对比了以下，发现作者你的onnx是官网基础上又加了一个很长的后处理对吗？所以如果我现在想用自己训练的模型去进行自动标注，得到的结果就明显不对。有什么比较快速的能让我使用自己的模型去做自动标注的方法吗，还是说我得自己修改插件AI_auto_annotate的代码？

[yatengLG]

star一个，我教你啊，哈哈哈哈哈哈

---

正经说：主要是导出的时候，需要指定一些参数。插件的readme有写。

https://github.com/yatengLG/ISAT_plugin_auto_annotate?tab=readme-ov-file#%E6%A8%A1%E5%9E%8B%E6%96%87%E4%BB%B6

[MichaelVi1866]

非常感谢大神！！！！！！！！
我还有个问题就是，为什么我把目标检测模型放到你的AI自动标注插件上，也可以做分割的检测呢？就是不管是目标检测模型还是分割模型，都可以用来做自动标注？

[yatengLG]

是的，不论是目标检测还是分割模型，都可以进行自动标注。

• 对于目标检测模型，ISAT会基于输出的预测框，使用SAM框提示，生成mask。
• 对于图像分割模型，ISAT也会会基于输出的预测框，使用SAM框提示，生成mask。
• 对于图像分割模型，没有直接使用输出的多边形，是为了接口统一，如果想使用输出的多边形进行提示，需要修改当前代码：1. 获取分割模型输出的多边形；2. 将多边形转换为masks；3. 使用ISAT提供的mask提示，优化mask；4. 或者也可以将2与3的步骤省略，直接在ISAT画布上添加多边形。