Faster RCNN training with matlab

1 次查看（过去 30 天）
akhil s anand
akhil s anand 2017-12-29
I have been trying to implement a custom network with Faster RCNN in MATLAB, I have specified the ExecutionEnvironment as gpu in the training options for training both the RPN as well as the CNN for classification. But the RPN training is not utilising my GPU, and after hours of training on CPU it throws an error saying array (24GB) exceeded maximum array size preferences. I have a 6 GB NVIDIA 1060 GPU in my PC and it was working well with MATLAB for my custom classification CNN training. But when I include the faster RCNN (RPN) it is not utilising the GPU. Please help to resolve the issue

请先登录，再回答此问题。

回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Object Detection 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by