yolov8目标检测+segment anything自动打实例分割标签_segment anything model 实现图像分割打标签

作者：正经夜光杯 | 2024-07-12 05:42:51

踩

segment anything model 实现图像分割打标签

最近出的segment anything只开放了预测功能，不能训练，拿自己的数据集测试，果然，不Fine tuning一下也是用不了的，不过看了分割效果还挺不错，用来自动打实例分割标签还真是不错的选择。

效果图

大致步骤就是1.用训练好的yolov8目标检测模型把图片的检测框记录下来；2.把图片和检测框文件放到SAM(segment anything model)里，然后和sam的demo里一样把图片和所有的检测框提示放进去,得出掩码；3.用opencv把掩码用多边形逼近得到点集合;4.把点集合按照labelme的json格式写到json文件中保存。有的掩码并不是非常好，尤其是多物体和花纹比较多的时候分割的并不是很好，还是得自己用labelme检查一遍，当然也许是因为我用的只是base模型，但是我的电脑运行不了(显卡太垃圾)large模型更不用说huge模型了。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/正经夜光杯/article/detail/812808