发现了一个好用的工具:
pycococreator工具,GitHub下载链接:https://github.com/waspinator/pycococreator
按教程安装
按照给定的格式重命名图片名称
有一个问题是,如果一张图中有多个类别的目标,需要有多个mask图
修改图片及标注的路径
运行
转换完json后,里面有测试代码,可以读取json里的标注看一下,是否正确,和cocoAPI里提供的差不多
主要还是通过opencv里的边缘检测方法,提取到mask的轮廓。
遇到的问题:
其中,在根据mask的png图片重新生成轮廓时,会出现很多小的圈,大多数为4个点,这些暴力剔除掉,具体是修改了轮廓点的个数阈值
/home/user/Downloads/pycococreator/examples/shapes/pycococreatortools/pycococreatortools.py中
def binary_mask_to_polygon(binary_mask, tolerance=0):函数中的 if len(contour) < 5
尽量还是直接标注成json输出吧!!!