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