如何在TensorFlow对象检测模型中生成模型摘要或绘制模型体系结构？ [英] How to generate a model summary or plot the model architecture in Tensorflow object detection models?

问题描述

是否可以查看模型摘要或绘制TensorFlow对象检测Git存储库中提供的TensorFlow对象检测模型。例如，查看更快的R-CNN模型的模型摘要？

推荐答案

可以使用模型摘要()Form TensorFlow查看模型摘要。 请参阅示例代码。

# Create the base model from the pre-trained model MobileNet V2
IMG_SHAPE = IMG_SIZE + (3,)
base_model = tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE,
                                               include_top=False,
                                               weights='imagenet')

# Let's take a look at the base model architecture
base_model.summary()

#输出

Model: "mobilenetv2_1.00_160"
__________________________________________________________________________________________________
Layer (type)                    Output Shape         Param #     Connected to                     
==================================================================================================
input_1 (InputLayer)            [(None, 160, 160, 3) 0                                            
__________________________________________________________________________________________________
Conv1 (Conv2D)                  (None, 80, 80, 32)   864         input_1[0][0]                    
__________________________________________________________________________________________________
bn_Conv1 (BatchNormalization)   (None, 80, 80, 32)   128         Conv1[0][0]                      
__________________________________________________________________________________________________
Conv1_relu (ReLU)               (None, 80, 80, 32)   0           bn_Conv1[0][0]                   
__________________________________________________________________________________________________
expanded_conv_depthwise (Depthw (None, 80, 80, 32)   288         Conv1_relu[0][0]                 
__________________________________________________________________________________________________

这篇关于如何在TensorFlow对象检测模型中生成模型摘要或绘制模型体系结构？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！