Проблема с обнаружением объектов в Faster RCNN для настраиваемых размерных изображений с плагином TensorRT - PullRequest
0 голосов
/ 29 июня 2019

Я поднимаю этот вопрос после двухдневных усилий, но не могу решить мою существующую проблему. У меня есть изображение 5000 * 600 , которое я хочу вывести с помощью Faster RCNN, используя плагин тензоррт.

Faster R-CNN принимает 3 канала 375x500 изображений в качестве ввода от официального документация . Я видел такое же измерение из файла . Перейти на номер строки 26 и 27 . Я попытался изменить размеры в строках с номерами 26 и 27 , чтобы они могли сделать вывод для настраиваемых форм изображения. Но такие усилия были бесполезны.

Я установил tenorrt в моей системе Ubuntu, используя официальную документацию Nvidia, и работает в / usr / src / tenorrt, где у меня есть примеры а также данные. Файл конфигурации для FasterRCNN ( sampleFasterRCNN.cpp ) можно найти в пути / usr / src / tenorrt / samples / sampleFasterRCNN.

Я столкнулся с другой проблемой. Если я собираюсь сделать вывод об одном изображении, т.е. использовать одно изображение в imagelist . Пожалуйста, посмотрите строку номер 294 в файле (std :: vector imageList = {"004545.ppm", "test12.ppm"};), где показана ошибка при сбросе ядра.

Вместо передачи одного изображения я передал несколько изображений, и такая ошибка (дамп памяти) устранена. Я не поняла почему?

Я хочу настроить Faster RCNN с настроенной формой изображения, используя плагин tenorrt. Есть ли способ сделать это. Я использую замороженный файл (.pb) из тензорного зоопарка. Пожалуйста, помогите мне.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...