RetinaNet(基于resnet50和fpn)的tensorboard网络架构图

采用网络的backbone部分,为了能运行tensorboard,所以必须要是完整的网络,所以结尾采用提取有效特征层P7(tensor类型),进行Flatten拉平,然后接了一层全连接,使用虚拟数据进行空跑,才能进入tensorboard。 Keras和TF是可以互通,使用tf.keras更加方便,中间可以嵌套tf,使用with tf.name_scope(‘stage5’): 可以将网络部分隐藏
相关文章
相关标签/搜索