想再多个分类中单独提取一个分类

[CYHooo]

导师你好， 我现在用mask-rcnn做训练的时候做了四个分类（2个不同种类的建筑和2个不同种类的道路）， 在可视化中想单独提取建筑类的结果。 尝试了直接减掉class_ids和用list的方式返回了class_names中我想要的分类，但是都失败了。 想问一下应该更改哪一部分可以做到单独提取我想要的分类？ 谢谢T^T

[bubbliiiing]

鹅，需要去detect image判断类别诶

[CYHooo]

鹅，需要去detect image判断类别诶

这一部分我直接重新训练了。。。detect_image那里我没找到idx的输入信号。。。

另外还想请教一下： 在mask_rcnn.py里有一个postprocess函数，在 #将语义分割结果映射到原图上# 那里会消耗将近75G的内存容量。（原图大小是10870*10870大小的.tif图片）有没有什么可以优化的办法减少内存消耗？

[bubbliiiing]

鹅，这个是绘图呀~你的图片这么大才会这样

[bubbliiiing]

这个已经是矩阵存储了，只是把矩阵映射到10870*10870这么大，这个无法避免

[bubbliiiing]

也许用int8 fp16这种方式存储可能可以

[CYHooo]

也许用int8 fp16这种方式存储可能可以

感谢解答！这部分我会做个尝试，现在还没完全搞懂函数内具体怎么运算矩阵进行的resize。

之前想法是把矩阵变成坐标，用坐标直接进行计算但是没能完成换算成坐标的那一步，先试试您说的换个存储方式，哈哈。

[CYHooo]

你好，我还有关于mAP计算的疑问。对于utils/utils_map.py这个文件，这里的函数是只提取predict之后信息，没有函数可以直接调用，用来计算mAP结果吗。。。。。

[bubbliiiing]

呐？不是用了coco的工具啊吗

[CYHooo]

呐？不是用了coco的工具啊吗

不好意思，json文件刚接触没多久有点看不懂这段代码T.T MAKE_JSON这个类是返回关于bbox和segm的json文件吗？我保存好pred的json文件，再调出groudtruth的json，然后计算mAP是不是可以.... 这里segmentation里的score是postprocess返回的mask_args变量吗？

[bubbliiiing]

https://github.com/bubbliiiing/mask-rcnn-tf2/blob/master/eval.py 这里有计算？你看看

[CYHooo]

https://github.com/bubbliiiing/mask-rcnn-tf2/blob/master/eval.py 这里有计算？你看看

感谢！！ 代码注释写的很完整，看完代码可以理解怎么运行的json文件！