Как привязать точки к краю во время обнаружения ориентира / ключевой точки с помощью сверточной нейронной сети
Моя текущая задача - обучить Conv-NN предсказывать ориентиры на полутоновом изображении.
Границы: некоторые ориентиры должны находиться точно на краю между объектом и черным фоном. Эту область легко определить по контуру.
Вопрос: Есть ли способ сделать так, чтобы ориентиры располагались точно на краю?
Абстрактный пример изображения. Красные точки - это целевые ориентиры.
Я могу представить одну вещь - добавить дополнительное измерение к входному изображению, содержащему только контур. Но я ожидаю, что CNN узнает об этом, если возможно.
Другая идея может заключаться в том, чтобы построить свою собственную метрику, каким-либо образом взаимодействуя с информацией о контурах или краях...
Что еще можно придумать, чтобы избавиться от этого страдания?