Пользовательская функция потерь в зависимости от y_true - PullRequest
Новая
       47

Пользовательская функция потерь в зависимости от y_true

0 голосов
/ 12 апреля 2020

Я пытаюсь настроить функцию потери 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()
...