将mmdet掩码r-cnn结果转换为coco数据集以提取分割# to检测。

Question

我得到了一个经过训练的.pth检查点文件和.py配置文件的掩膜r-cnn模型，并且我能够使用普通教程在视频上执行对象分割。model.html，我正在尝试将分割结果转换为json格式的coco格式，这样我就可以将它插入到管道中。要指定:我需要将推理检测器的结果转换为coco格式的json文件。来自mmdetection检测github repo的演示/视频演示代码：

​

结果将是：

​

我在这里找到了一个可能的解决方案：https://mmdetection.readthedocs.io/en/latest/api.html#module-mmdet.datasets。其中，results2json函数将转储分段的结果(+ bbox等)到coco文件，但是我应该创建一个CocoDataset实例来使用这个函数。我没有带有注释的json文件，更不用说对线了吗？(检查参数以创建一个参数)。

如何使用分割的结果生成coco格式的json文件？提前谢谢。

Answer 1

您可以通过以下方式轻松地生成数据集

dataset = build_dataset(cfg.data.test) 

在构建cfg时，可以插入测试管道和配置的工作方式。

检查这里中的cfg.data.test内容

然后，您可以使用dataset.result2json(result)获得json格式的注释。