Q&A
前提・実現したいこと
切り出した顔画像(jpg512ピクセル正方形)から、性別、年齢をローカルで判定したいです
サンプルPythonプログラムにtensorflow-gpuを使用している物がありました
forループで、1回づつ呼び出しているので、tensorflow-gpuを使用しても遅いです
CPUで実行するにはコードをどう書き換えたらいいでしょうか？
GPUだと大量の顔画像を処理していると、しばらくするとOOMエラーが発生します
発生している問題・エラーメッセージ
OOMエラー、CUDAエラー
該当のソースコード
python
1with tf.Graph().as_default(): 2 gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction) 3 sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False)) 4 with sess.as_default(): 5 pnet, rnet, onet = align.detect_face.create_mtcnn(sess, None)
補足情報（FW/ツールのバージョンなど）
python-3.7.6
tensorflow (1.13.2)
tensorflow-gpu (1.13.2)
あなたの回答
tips
プレビュー
