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

Question

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

Answer 1

在張量流中，您需要爲兩個模型明確指定the computational graph。

# 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) 

在一個應用程序中構建多個圖形時還有一些注意事項，請參見this question。