python - 用于灰度图像的预训练对象检测 API 模型（1 通道）

Question

我想通过 Tensorflow 的对象检测 API 在 CT 扫描图像（灰度图像）中检测特征，并可能对其进行分割。所以我有两个问题：

- 对象检测 API 中是否有任何预训练模型将灰度图像作为微调训练的输入？

- 或者，我如何训练现有的模型，即用于 RGB 图像，以便我可以用灰度图像训练它们？应该在代码中进行哪些修改？

标签： pythontensorflowcomputer-visionobject-detection

Answer 1

以防万一有人对我如何解决这个问题感兴趣：

对于灰度图像的分割，我发现了这个架构：U-Net The tensorflow adapt by Akeret et al。可以在Github - UNet中找到

最后我没有使用 API 对象检测，因为我不确定预训练模型在 1 通道图像中的性能如何，因为它们是为 RGB 图像创建的。