Caffe bvlc_googlenet минимально допустимые размеры

Question

Какой минимальный размер изображения принимается моделью bvlc_googlenet, реализованной Caffe?

Я использую 50 x 50 изображений с crop_size = 36 , где я получаю следующую ошибку при запуске решателя:

caffe::Blob<>::Reshape() - Floating point exception

Мне нужно изменить размеры моих изображений до 256 x 256 (размер по умолчанию для модели bvlc_googlenet) с crop_size = 224 , чтобы избежать ошибки.

Эта модель принимает только размеры по умолчанию, или мне нужно немного взломать, чтобы это произошло?

Спасибо !!

Answer 1

После нескольких часов попыток решить проблему, я понял, с чем столкнулся. GoogleNet принимает 224 * 224 изображения в качестве входных данных по умолчанию, так как оно очень глубокое и после набора слоев свертки и объединения, используя 50 * 50 Изображения (или 36 * 36 после кадрирования) приведут к выводу очень маленького размера после передачи ввода в несколько слоев, меньших, чем размер ядра следующего слоя.Это приведет к исключению Reshape, похожему на то, с которым я столкнулся здесь.

Решение:

1. Хотя редактировать не рекомендуется.kernel_size параметр уровня, вызывающего исключение (чтобы продолжать работу в соответствии со спецификациями NN), это решит проблему, где вы можете выбрать меньший размер ядра и затем тестировать результаты, пока он не заработает.

2. Соответствует стандартным GoogleNet спецификациям, изменяя размер входных изображений до 254 * 254 ( с сохранением размера кадрирования до 224 ) или напрямуюизменив его на 224 * 224 и удалив параметр crop_size .