Тренируйтесь по собственному набору данных.Mask_RCNN Ресурс исчерпан: OOM при распределении

Question

Я пытаюсь обучить свой собственный набор данных с помощью Mask_RCNN, но я получаю следующие ошибки структуры: Я следовал учебнику github в Mask R-CNN для обнаружения объектов.

Могу ли я сделать что-нибудь, чтобы уменьшить объем памяти, необходимый для обучения набора данных?Или как я могу решить эту проблему?

Answer 1

Требуется больше информации о вашем ПК. Я думаю, что, возможно, из-за того, что размер пакета или размер изображения слишком велик, ему требуется больше видеопамяти или памяти.-models / object_detection, GTX1080TI 11 ГБ работает нормально. (batch_size = 1, image_size = 1000 * 600).