изменение ssd net в тензорном потоке - PullRequest
0 голосов
/ 17 января 2020

Если бы я использовал Keras, модификация архитектуры была бы прямой модификацией сетевых слоев:

        x = Conv2D(32, (3, 3), padding="same")(inputs)
        x = Activation("relu")(x)
        x = Conv2D(32, (3, 3), padding="same")(x)
        x = Activation("relu")(x)
        x = MaxPooling2D(pool_size=(2, 2), strides=(2, 2))(x)

Но я не могу найти эту структуру в Tensorflow:

https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/ssd_mobilenet_v2_coco.config

Я смотрю на нужный файл? Другими словами, как мне go добавить дополнительные слои (будь то сверточный, maxpool или полностью связанный) в мою модель тензорного потока?

1 Ответ

0 голосов
/ 18 января 2020

Похоже, что фактическая модель была абстрагирована до
https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py

Хотя я мог бы предположить, что должен быть более простой способ изменить архитектуру нейронный net, чем просто изменить этот файл ...

...