В развертывании сети отсутствует слой Softmax

Question

Я пытаюсь использовать предварительно подготовленную модель (VGG 19) для DIGITS, но я получил эту ошибку.

ERROR: Your deploy network is missing a Softmax layer! Read the documentation for custom networks and/or look at the standard networks for examples

Я пытаюсь проверить с помощью своего набора данных, который имеет только два класса.

Я прочитал this и this попытаюсь изменить последний слой, но также я получил ошибку. Как изменить слои на основе нового набора данных?

Я пытаюсь изменить последний слой и я получил ошибку

ERROR: Layer 'softmax' references bottom 'fc8' at the TRAIN stage however this blob is not included at that stage. Please consider using an include directive to limit the scope of this layer.

Answer 1

у вас возникли проблемы, потому что вы пытаетесь загрузить сеть «поезд/VAL», когда вам действительно нужно быть загрузив сеть «все-в-одном». К сожалению, мы не документируем это очень хорошо. Я создал an RFE, чтобы напомнить нам об улучшении документации.

Попробуйте настроить последние слои в сети, чтобы выглядеть примерно так: https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

Для получения дополнительной информации, here как я предложила обновить примеры сетей Caffe к всем-в-одном сеток, и here является как я обновил сети DIGITS по умолчанию, чтобы быть сетями «все-в-одном».