0

如果我使用 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

我在看正确的文件吗?换句话说,我将如何向我的 tensorflow 模型添加额外的层(无论是卷积层、最大池层还是全连接层)?

4

1 回答 1

0

看起来实际模型已被抽象为
https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py

虽然我想必须有一种更简单的方法来改变神经网络的架构,而不仅仅是修改那个文件......

于 2020-01-18T15:20:34.820 回答