tensorflow 当mobilenetv2瓶颈块中的输入层和最后一层大小几乎不一样时,你如何添加它们呢?

busg9geu  于 2022-12-27  发布在  其他
关注(0)|答案(1)|浏览(87)

我在理解mobilenetv 2架构的瓶颈块的基本思想上有很多困难。对于stride = 1,你把输入层加到最后一层(又名跳过连接),这些实际上从来不是相同的大小!你应该怎么加到几乎从来不是相同大小的Tensor上呢?
我试着从很多方面考虑它,但我看不出在跨距=1时会发生什么。

ct2axkht

ct2axkht1#

你可以使用model. add添加层

import tensorflow as tf

base_model = tf.keras.applications.MobileNetV2(input_shape=(96,96,3),
                                               include_top=False,
                                               weights='imagenet')

model = tf.keras.Sequential()
model.add(base_model)
model.add(tf.keras.Input(shape=( )))

谢谢你。

相关问题