Форма цели в пользовательской функции потерь - PullRequest
0 голосов
/ 09 марта 2020

Я пишу более быструю реализацию R-CNN с тензорным потоком 2.1. Форма моего выходного тензора (1, 32, 32, 9, 5) и форма цели (1, None, 5) - это ограничивающие прямоугольники, число которых не является постоянным.

Я хочу написать пользовательскую функцию потерь, и когда я запускаю

model.compile(optimizer='adam', loss=my_loss_function)

in my_loss_function я получаю форму цели (None, None, None, None, None). Я хочу установить форму цели на (1, None, 5), как я могу это сделать?

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