Как передать небольшие вырезанные части большого изображения для обучения CNN для обнаружения и локализации объектов (спутниковые изображения)

Я пытаюсь реализовать обнаружение объектов на спутниковых снимках. У меня есть аннотированный набор данных, но изображения большие, и модель принимает только входные данные размером 416 x 416. Как я могу передать небольшие части изображения в сети, гарантируя, что аннотации сохраняются. Кроме того, как объединить эти результаты во время теста?

1 ответ

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

https://github.com/lopuhin/kaggle-dstl/blob/master/train.py

Другие вопросы по тегам