summaryrefslogtreecommitdiff
path: root/tools/python
diff options
context:
space:
mode:
authorLimin Wang <lance.lmwang@gmail.com>2021-04-28 18:47:08 +0800
committerGuo, Yejun <yejun.guo@intel.com>2021-04-29 20:02:29 +0800
commitf183d6555e714e00b41aec728feb8a731826cbdc (patch)
treeec784e1efd0d7819d4a50338679c6c3e01eeb21d /tools/python
parent4e8d22478bc854949a9adbda97348f2997e1dc8d (diff)
avfilter/dnn/dnn_backend_tf: simplify the code with ff_hex_to_data
please use tools/python/tf_sess_config.py to get the sess_config after that. note the byte order of session config is in normal order. bump the MICRO version for the config change. Signed-off-by: Limin Wang <lance.lmwang@gmail.com>
Diffstat (limited to 'tools/python')
-rw-r--r--tools/python/tf_sess_config.py45
1 files changed, 45 insertions, 0 deletions
diff --git a/tools/python/tf_sess_config.py b/tools/python/tf_sess_config.py
new file mode 100644
index 0000000000..5705b1ab9d
--- /dev/null
+++ b/tools/python/tf_sess_config.py
@@ -0,0 +1,45 @@
+# Copyright (c) 2021
+#
+# This file is part of FFmpeg.
+#
+# FFmpeg is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# FFmpeg is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with FFmpeg; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+# ==============================================================================
+
+# verified with Python 3.6.8 on CentOS 7.2
+import tensorflow as tf
+
+visible_device_list = '0' # use , separator for more GPUs like '0, 1'
+per_process_gpu_memory_fraction = 0.9 # avoid out of memory
+intra_op_parallelism_threads = 2 # default in tensorflow
+inter_op_parallelism_threads = 5 # default in tensorflow
+
+gpu_options = tf.compat.v1.GPUOptions(
+ per_process_gpu_memory_fraction = per_process_gpu_memory_fraction,
+ visible_device_list = visible_device_list,
+ allow_growth = True)
+
+config = tf.compat.v1.ConfigProto(
+ allow_soft_placement = True,
+ log_device_placement = False,
+ intra_op_parallelism_threads = intra_op_parallelism_threads,
+ inter_op_parallelism_threads = inter_op_parallelism_threads,
+ gpu_options = gpu_options)
+
+s = config.SerializeToString()
+# print(list(map(hex, s))) # print by json if need
+
+print('a serialized protobuf string for TF_SetConfig, note the byte order is in normal order.')
+b = ''.join(format(b,'02x') for b in s)
+print('0x%s' % b) # print by hex format