本博客试图解决以下问题
完整代码可以直接下载使用:GitHub地址
- labelme的标注格式转到COCO的格式。 (对应labelme2coco.py)
- 从COCO的instance segmentation 得到每一个instance 的 binary mask。(对应make_list.py)
- 对binary mask resize 到指定大小 (并去掉那些比较小的mask)(对应change_size.py)
一些要求
- 安装labelme
- 安装COCO dataset的依赖
- 以及其他依赖,如cv2。
从labelme到COCO
要准备的东西:
- 已经用labelme标注好的文件,放在src_dataset文件夹下。
- 准备labels.txt文件,前面的两个需要固定,否则会出错,后面的根据你自己的情况来,可以见给出的例子里面。
- 在labelme2coco.sh中指定好文件夹名字
- 执行该文件(请先安装好labelme):sh labelme2coco.sh