Какой размер изображения подходит для MaskRCnn? - PullRequest
1 голос
/ 06 ноября 2019

Я тренирую MaskRcnn на изображениях документов с размером (256,256,3) и не могу получить хорошую точность. если я увеличу размер изображения, то код выдаст ошибку чайки памяти графического процессора. Должен ли я увеличить тренировочный имидж? Что еще я могу сделать, чтобы улучшить точность.

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

1 Ответ

0 голосов
/ 13 ноября 2019

Поскольку размер вашего изображения небольшой, попробуйте использовать модель SSD вместо модели maskrcnn, это также даст лучшее время обнаружения.

...