Ошибка пакета обнаружения объектов Tensorflow при изменении размера пакета

Я пробую недавно выпущенный API обнаружения объектов для tensorflow. Я использовал пример программы обучения в учебнике, т. Е. Тонкую настройку модели FRCNN-Resnet на наборе данных домашних животных. При использовании только одного графического процессора всегда появляется сообщение об ошибке, когда я изменяю размер пакета на значение больше 1 (по умолчанию 1). Сообщение об ошибке выглядит так:

InvalidArgumentError (обратную трассировку см. выше): ConcatOp: размеры входных данных должны совпадать: shape[0] = [1,750,600,3] vs. shape[1] = [1,600,804,3]


person Xingdi    schedule 06.11.2017    source источник


Ответы (1)


Вероятно, вы используете keep_aspect_ratio_resizer, который позволяет каждому изображению иметь разный размер, а это означает, что вы можете тренироваться только с размером пакета 1. Для обучения с большим размером пакета в настоящее время единственный способ справиться с этим в API — использовать fixed_shape_resizer . Примеры см. в некоторых конфигурациях SSD.

person Jonathan Huang    schedule 06.11.2017