Как изменить структуру модели Mobilnet-SSD в API обнаружения объектов Tensorflow?

Question

Я тренируюсь Mobilnet-SSD для обнаружения маленьких голов.
Поэтому я хочу добавить немного box_predictor_layer перед слоем сети, например добавить box_predictor_layer для подключения 5/6-го слоя свертки.

Как это сделать?
Я читаю все файлы .proto, нахожу ssd.proto и ssd_anchor_generator.proto , box_predictor.proto .
Но я не знаю, как их изменить, потому что график тензорной доски настолько беспорядочный.

Answer 1

Вам необходимо добавить еще одну «точку выхода» (т. Е. Карту объектов, к которой применяется обнаружение) из магистрали. Сделайте это, добавив имя слоя в здесь и -1 в соответствующем месте в layer_depth (это означает, что вам не нужно указывать глубину карты объектов, так как это дано позвоночником). Если вам нужно только обнаружить маленькие головки, поэтому вам не нужно обнаруживать большие, вы, вероятно, можете избавиться (по крайней мере, от некоторых) от дополнительных карт функций SSD, которые добавляются после магистрали с помощью 's in from_layer и 512, 256, 256, 128 в layer_depth. Не забудьте изменить num_layers в конфигурационном файле соответственно:

model {
  ssd {
    anchor_generator {
      ssd_anchor_generator {
        num_layers: <num_of_feature_maps>
        ...
      }
    }
  }
}