用Yolact模型训练自己的数据集

1.首先用labelme标注数据(分割法)

2.yolact训练的数据集为coco数据集,所以要转化数据集,使用从Git下载的labelme项目中的labelme2coco转换就可以

这里因为我的标注里有circle,刚开始训练总是失败报错,因为多边形标注至少要四个点的信息,labelme2coco中需要增加一段让它支持圆形标注

 Git中有解决方法https://github.com/wkentaro/labelme/issues/537

这里也放出增加的一段代码(Git里的方法)

            elif shape_type == "circle":

                # Create polygon shaped based on connecting lines from/to following degress
                bearing_angles = [0, 15, 30, 45, 60, 75, 90, 105, 120, 135, 150, 165,
                180, 195, 210, 225, 240, 255, 270, 285, 300, 315, 330, 345, 360]

                orig_x1 = points[0][0]
                orig_y1 = points[0][1]

                orig_x2 = points[1][0]
        
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值