MXnet - ImageRecordIter и расширение данных для включения пула ROI CNN

Question

Как я могу выполнить увеличение данных, когда я использую ROI-Pooling в сети CNN, которую я разработал с использованием MXnet?

Например, предположим, у меня есть архитектура resnet50, которая использует слой roi-pooling, и я хочуиспользуйте случайное увеличение данных в итераторе ImageRecord.

Существует ли автоматический способ, с помощью которого координаты данных в роусе передаются в слой пула рои, преобразуются для применения в изображениях, генерируемых процессом дополнения данныхитератора ImageRecord?

Answer 1

Вы должны быть в состоянии повторно использовать ImageDetRecordIter для этого.Он предназначен для использования с данными Обнаружения объектов, содержащими ограничивающие блоки, но вы можете определить ограничивающие блоки в качестве ваших ROI.И теперь, когда вы применяете операции увеличения (такие как переворот и вращение), координаты ограничивающих рамок будут корректироваться в соответствии с изображениями.

В противном случае вы можете легко написать свои собственные transform работает с использованием Gluon и может использовать любое расширение OpenCV для применения как к вашему изображению, так и к областям интереса.Просто напишите функцию, которая принимает данные и метку и возвращает расширенные данные и метку.