libavfilter/dnn: add header into native model file

Guo, Yejun committed 7 years ago

Commit 022f50d1 parent 3b3150c3 files · +70 −2Browse files

Showing the changes to tools/python (2 of 3 files). Show all

Unified · Split

2 files

 import tensorflow as tf
 import numpy as np
 import sys, struct
+import convert_header as header
 __all__ = ['convert_from_tensorflow']
     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)

@@ -0,0 +1,26 @@
	1	# Copyright (c) 2019
	2	#
	3	# This file is part of FFmpeg.
	4	#
	5	# FFmpeg is free software; you can redistribute it and/or
	6	# modify it under the terms of the GNU Lesser General Public
	7	# License as published by the Free Software Foundation; either
	8	# version 2.1 of the License, or (at your option) any later version.
	9	#
	10	# FFmpeg is distributed in the hope that it will be useful,
	11	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	# Lesser General Public License for more details.
	14	#
	15	# You should have received a copy of the GNU Lesser General Public
	16	# License along with FFmpeg; if not, write to the Free Software
	17	# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	18	# ==============================================================================
	19
	20	str = 'FFMPEGDNNNATIVE'
	21
	22	# increase major and reset minor when we have to re-convert the model file
	23	major = 0
	24
	25	# increase minor when we don't have to re-convert the model file
	26	minor = 1

Keyboard shortcuts