diff options
| author | Guo, Yejun <yejun.guo@intel.com> | 2019-09-02 12:35:58 +0800 |
|---|---|---|
| committer | Pedro Arthur <bygrandao@gmail.com> | 2019-09-04 11:13:21 -0300 |
| commit | 022f50d3febe5232e67d38b0292d86f7aa8a360c (patch) | |
| tree | f9cdd9a5cde14a0c2e175a5a85fa54176c68ff69 /tools/python/convert_from_tensorflow.py | |
| parent | 3b3150c45f1ebb3635e55e76b63439d8d62de85f (diff) | |
| download | ffmpeg-022f50d3febe5232e67d38b0292d86f7aa8a360c.tar.gz | |
libavfilter/dnn: add header into native model file
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'tools/python/convert_from_tensorflow.py')
| -rw-r--r-- | tools/python/convert_from_tensorflow.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tools/python/convert_from_tensorflow.py b/tools/python/convert_from_tensorflow.py index bab11a5aab..1437ad358d 100644 --- a/tools/python/convert_from_tensorflow.py +++ b/tools/python/convert_from_tensorflow.py @@ -20,6 +20,7 @@ import tensorflow as tf import numpy as np import sys, struct +import convert_header as header __all__ = ['convert_from_tensorflow'] @@ -229,6 +230,8 @@ class TFConverter: def dump_to_file(self): with open(self.outfile, 'wb') as f: + f.write(header.str.encode('utf-8')) + np.array([header.major, header.minor], dtype=np.uint32).tofile(f) self.dump_layers_to_file(f) self.dump_operands_to_file(f) np.array([self.layer_number, len(self.name_operand_dict)], dtype=np.uint32).tofile(f) |