#python #tensorflow #backpropagation #bounding-box #shapely
Вопрос:
Я пытаюсь написать пользовательскую функцию потерь в tensorflow,которая штрафует долговые расписки двух повернутых (не обязательно выровненных по оси) ограничивающих рамок,заданных (x,y,a,w, h), где (x, y) — координаты центра ограничивающей рамки, а » a » — угол поворота от 0 до 360 градусов. Я хотел бы сделать это с помощью функций тензорного потока, чтобы у меня была дифференцируемая функция потерь, которую можно использовать в backprop.
В настоящее время я знаю, как легко найти долговые расписки повернутых ограничивающих прямоугольников с помощью Shapely, но я не уверен, работает ли вычисление формы с использованием Polygon.intersection в функции потерь тензорного потока. Мои вопросы таковы:
- Могу ли я использовать shapely в пользовательской функции потерь тензорного потока для обучения регрессии ограничивающего прямоугольника?
- Если нет, то как бы я вычислил долговую расписку для повернутых ограничивающих рамок с использованием TF?