apps/ffmpeg-rockchip
1
0
Fork 0
mirror of https://github.com/nyanmisaka/ffmpeg-rockchip.git synced 2025-10-06 01:07:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

libavfilter/dnn: add header into native model file

Browse Source 
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
This commit is contained in:
Guo, Yejun
2019-09-02 12:35:58 +08:00
committed by Pedro Arthur
parent 3b3150c45f
commit 022f50d3fe
3 changed files with 70 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tools/python/convert_from_tensorflow.py
View File

@@ -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)