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]