From 022f50d3febe5232e67d38b0292d86f7aa8a360c Mon Sep 17 00:00:00 2001 From: "Guo, Yejun" Date: Mon, 2 Sep 2019 12:35:58 +0800 Subject: libavfilter/dnn: add header into native model file Signed-off-by: Guo, Yejun Signed-off-by: Pedro Arthur --- tools/python/convert_from_tensorflow.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'tools/python/convert_from_tensorflow.py') diff --git a/tools/python/convert_from_tensorflow.py b/tools/python/convert_from_tensorflow.py index bab11a5..1437ad3 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) -- cgit v1.1