卷積神經網絡-如何在同一張圖像上使用兩個cnn模型

Question

我想在同一張圖片上使用兩個訓練有素的CNN（卷積神經網絡）模塊。 我訓練了兩個模塊，一個用於檢測，一個用於分類。 現在，我想在同一圖像上使用這兩個模塊。 該代碼在python中使用keras和tensorflow庫。 同一張圖片上有兩個不同的CNN

Answer 1

在tensorflow中，您需要為兩個模型明確指定計算圖 。

# build two separate graphs `g1` and `g2`

tf.reset_default_graph()
with tf.Session(graph=g1) as session:
    result = sess.run(detect, feed_dict={x: test_x})
    print(result)

tf.reset_default_graph()
with tf.Session(graph=g2) as session:
    result = sess.run(recognize, feed_dict={x: test_x})
    print(result)

在一個應用程序中構建多個圖形時，也有一些警告，請參見此問題 。