Я пишу более быструю реализацию 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)
, как я могу это сделать?