【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别

首先是数据集的准备，这里以猫狗识别为例子

使用labelme对图片进行打标签，生成json格式的图片，而且labelme的版本尽量新一点，老版本 比如3.16.7版本 对后面生成SegmentationClassPNG 和 SegmentationObjectPNG会有问题。

直接在命令行输入labelme即可

labelme

 在使用labelme打上标签之后，就会在本地生成json格式

然后我们在使用代码：labelme2voc.py，将其从json格式转成SegmentationClassPNG 和 SegmentationObjectPNG格式。

代码如下：

from __future__ import print_function

import argparse
import glob
import os
import os.path as osp
import sys

import imgviz
import numpy as np

import labelme


def main():
    p