RetinaNet(基於resnet50和fpn)的tensorboard網絡架構圖

採用網絡的backbone部分,爲了能運行tensorboard,所以必須要是完整的網絡,所以結尾採用提取有效特徵層P7(tensor類型),進行Flatten拉平,然後接了一層全連接,使用虛擬數據進行空跑,才能進入tensorboard。 Keras和TF是可以互通,使用tf.keras更加方便,中間可以嵌套tf,使用with tf.name_scope(‘stage5’): 可以將網絡部分隱藏
相關文章
相關標籤/搜索