42 lines
1.5 KiB
Python
Executable File
42 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# ==========================================================================================
|
|
# Copyright (c) 2020 AIIT XUOS Lab
|
|
# XiOS is licensed under Mulan PSL v2.
|
|
# You can use this software according to the terms and conditions of the Mulan PSL v2.
|
|
# You may obtain a copy of Mulan PSL v2 at:
|
|
# http://license.coscl.org.cn/MulanPSL2
|
|
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
# See the Mulan PSL v2 for more details.
|
|
|
|
# @file: mnist-c-model.py
|
|
# @brief: open file path and load model
|
|
# @version: 1.0
|
|
# @author: AIIT XUOS Lab
|
|
# @date: 2021/4/30
|
|
# ==========================================================================================
|
|
|
|
|
|
#tflite_file_path = 'mnist-default-quan.tflite'
|
|
tflite_file_path = 'mnist.tflite'
|
|
model_file_path = 'model.h'
|
|
|
|
tflite_file = open(tflite_file_path, 'rb')
|
|
tflite_data = tflite_file.read()
|
|
tflite_file.close()
|
|
tflite_array = [ '0x%02x' % byte for byte in tflite_data ]
|
|
|
|
model_content = '''unsigned char mnist_model[] = {
|
|
%s
|
|
};
|
|
unsigned int mnist_model_len = %d;
|
|
'''
|
|
# 12 bytes in a line, the same with xxd
|
|
bytes_of_line = 12
|
|
model_data = (',\n ').join([ (', ').join(tflite_array[i:i+bytes_of_line]) for i in range(0, len(tflite_array), bytes_of_line) ])
|
|
model_file = open(model_file_path, 'w')
|
|
model_file.write(model_content % (model_data, len(tflite_array)))
|
|
model_file.close()
|
|
|