【发布时间】:2020-03-02 20:16:21
【问题描述】:
我正在对大小为 (256,256,3) 的文档图像训练 MaskRcnn,但无法获得良好的准确度。如果我增加图像大小,代码会抛出 GPU 内存海鸥错误。我应该增加训练图像吗?我还能做些什么来提高准确性。
Training set size = 1000 images
Classes = 45
GPU = Nvidia GTX 1060
Actual Image resolution = 9000 x 6600
【问题讨论】:
-
我认为您必须提高训练图像的分辨率,因为您的实际图像分辨率太大而无法调整大小(256,256,3)。如果您的 GPU 内存没有足够的空间进行训练,如何更改批量大小?
标签: python keras deep-learning object-detection faster-rcnn