Я пытаюсь настроить функцию потери IoU. У моей модели 5 выходов. Первый - 0 или 1. Если в кадре есть объект, это 1, в противном случае - 0. Остальные 4 выхода являются ограничивающим прямоугольником (регрессия) Для каждого из y_true, если первый выход равен 1 из y_true, то это будет рассчитывать долговую расписку. В противном случае он не будет рассчитывать IoU. Я не мог перебрать тензоры y_true / y_pred. Как мне этого добиться?
def custom_loss(y_true, y_pred):
for y in y_true:
if first_index_value_is_iou:
calculate_IoU()