Требуемые потери выходного слоя не отображаются при печати кофе - PullRequest
Новая
       13

Требуемые потери выходного слоя не отображаются при печати кофе

0 голосов
/ 09 июля 2019

Я настраиваю существующую сеть, добавляя один собственный выходной слой. Используемая сеть уже настроена для распознавания лиц и извлечения черт лица.

Мой добавленный слой добавляется только к последней части (я надеюсь, что там еще достаточно данных), пытаясь извлечь угол рыскания.

Конец файла prototxt до моих изменений: 

layer {
  name: "cls_prob"
  type: "Softmax"
  bottom: "fc6_3"
  top: "cls_prob"
}
layer {
  name: "bbox_reg_3"
  type: "InnerProduct"
  bottom: "fc5_3"
  top: "bbox_reg_3"
  inner_product_param {
    num_output: 3
  }
}
layer {
  name: "rotate_reg_3"
  type: "InnerProduct"
  bottom: "fc5_3"
  top: "rotate_reg_3"
  inner_product_param {
    num_output: 1
  }
}

Тот же сегмент с моими изменениями + тренировочные изменения: 

layer {
  name: "cls_prob"
  type: "Softmax"
  bottom: "fc6_3"
  top: "cls_prob"
}
layer {
  name: "bbox_reg_3"
  type: "InnerProduct"
  bottom: "fc5_3"
  top: "bbox_reg_3"
  param {
    lr_mult: 0
    decay_mult: 0
  }
  param {
    lr_mult: 0
    decay_mult: 0
  }
  inner_product_param {
    num_output: 3
  }
}
layer {
  name: "rotate_reg_3"
  type: "InnerProduct"
  bottom: "fc5_3"
  top: "rotate_reg_3"
  param {
    lr_mult: 0
    decay_mult: 0
  }
  param {
    lr_mult: 0
    decay_mult: 0
  }
  inner_product_param {
    num_output: 1
  }
}
layer {
  name: "yaw_reg_3"
  type: "InnerProduct"
  bottom: "fc5_3"
  top: "yaw_reg_3"
  inner_product_param {
    num_output: 1
  }
}
layer {
  name: "loss"
  type: "EuclideanLoss"
  bottom: "yaw_reg_3"
  bottom: "label"
  top: "loss"
}

Я добавил yaw_reg_3, но вижу только распечатки потерь rotate_reg_3, bbox_reg_3 и cls_prob.

Спасибо

1 Ответ

0 голосов
/ 10 июля 2019

После отладки самого caffe с помощью gdb: после прохождения всех слоев, если слой не отображается как нижний слой для другого слоя, он определяется как выходной слой.Поскольку "yaw_reg_3" является нижней частью для "потерь", технически это не выходной слой, поэтому печатается только потеря для него, а не сам прогноз.

...