Detectron2 Mask R-CNN 细胞分割 - 无可见

问题描述

我是深度学习的新手,并尝试使用 Detectron2 Mask R-CNN 进行细胞分割。 我使用来自 http://celltrackingchallenge.net/2d-datasets/ - HL60 细胞的模拟核 - 训练数据集的图像和蒙版图像。我使用的文件夹是 here

我尝试按照 detectron2 colab tutorial 中的气球数据集格式创建和注册新数据集。 我有 1 个类,“细胞”。

我的问题是,在我训练模型后,在可视化预测时没有可见的掩码。也没有边界框或预测分数。 可视化的带注释的图像类似于 this,但预测的蒙版图像只是像 this 这样的黑色背景。

我可能做错了什么?我制作的 colab 是 here

解决方法

我通过使用 Matterport Mask R-CNN 和样本核数据集找到了当前的解决方法:https://github.com/matterport/Mask_RCNN/tree/master/samples/nucleus