yolov8目标检测+segment anything自动打实例分割标签

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

效果图

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

  • 7
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值