現在画像の深度計算(githubのリンク)を自分の関数に取り入れようとしています。
メイン関数のpredict.py内で深度計算をしている部分はなんとなく分かったのですが、それを自分の方に入れようとするとGPUの立ち上がりのメッセージが出たまま止まってしまいます。
原因が分からず困っております。多分tf.Sessionからは逃れられないのでどのようにすればいいか教えていただけると助かります。よろしくお願いいたします。
[元の関数内のmodelをメインで作っているところ]
input_node = tf.placeholder(tf.float32, shape=(None, height, width, channels))
net = models.ResNet50UpProj({'data': input_node}, batch_size, 1, False)
with tf.Session() as sess:
---print('Loading the model')
---net.load(model_data_path, sess)
---pred = sess.run(net.get_output(), feed_dict={input_node: img})
[自作コードへの組み込み(組み込み方が下手です)]
input_node = tf.placeholder(tf.float32, shape=(None, height, width, channels))
net = FCRN.tensorflow.models.ResNet50UpProj({'data': input_node}, batch_size, 1, False)
while True:
---frame = mymodule.video2jpg(1, count, tstr + orgstr)
---result = yolo.detect_image(frame)
---frame = np.array(frame).astype('float32')
---frame = np.expand_dims(np.asarray(frame), axis = 0)
---with tf.Session() as sess:
-------net.load("./FCRN/tensorflow/NYU_ResNet.npy", sess)
-------pred = sess.run(net.get_output(), feed_dict={input_node: frame})
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。