python - 什么样的图像尺寸适合 MaskRCnn？

Question

我正在对大小为 (256,256,3) 的文档图像训练 MaskRcnn，但无法获得良好的准确性。如果我增加图像大小，代码会抛出 GPU 内存海鸥错误。我应该增加训练图像吗？我还能做些什么来提高准确性。

Training set size = 1000 images
Classes = 45
GPU = Nvidia GTX 1060
Actual Image resolution = 9000 x 6600

Answer 1

由于您的图像尺寸很小，请尝试使用 SSD 模型而不是 maskrcnn 模型，这也将提供更好的检测时间。