mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-02 10:17:54 +00:00
f91dd916e5
build(codegen): suppress noise in console output
Redirect output from the code_generator binary so that when it's
used within the build system, it doesn't print unexpected,
distracting noise to the console. Generally, compiler or
generator commands don't print output unless there's an error.
This reverts "build(codegen): suppress noise in console output
(#2708)", commit d2495773, which attempted to fix the same
problem by adding a --quiet flag. A subsequent upgrade to the
Tensorflow Python package caused new noise that wasn't possible
to suppress with the flag.
BUG=description
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
""" Generates C/C++ source code capable of performing inference for a model. """
|
|
|
|
import os
|
|
|
|
from absl import app
|
|
from absl import flags
|
|
from collections.abc import Sequence
|
|
|
|
from tflite_micro.codegen import inference_generator
|
|
from tflite_micro.codegen import graph
|
|
from tflite_micro.tensorflow.lite.tools import flatbuffer_utils
|
|
|
|
# Usage information:
|
|
# Default:
|
|
# `bazel run codegen:code_generator -- \
|
|
# --model=</path/to/my_model.tflite>
|
|
# Output will be located at: /path/to/my_model.h|cc
|
|
|
|
_MODEL_PATH = flags.DEFINE_string(name="model",
|
|
default=None,
|
|
help="Path to the TFLite model file.",
|
|
required=True)
|
|
|
|
_OUTPUT_DIR = flags.DEFINE_string(
|
|
name="output_dir",
|
|
default=None,
|
|
help="Path to write generated source to. Leave blank to use 'model' path.",
|
|
required=False)
|
|
|
|
_OUTPUT_NAME = flags.DEFINE_string(
|
|
name="output_name",
|
|
default=None,
|
|
help=("The output basename for the generated .h/.cc. Leave blank to use "
|
|
"'model' basename."),
|
|
required=False)
|
|
|
|
|
|
def main(argv: Sequence[str]) -> None:
|
|
output_dir = _OUTPUT_DIR.value or os.path.dirname(_MODEL_PATH.value)
|
|
output_name = _OUTPUT_NAME.value or os.path.splitext(
|
|
os.path.basename(_MODEL_PATH.value))[0]
|
|
|
|
model = flatbuffer_utils.read_model(_MODEL_PATH.value)
|
|
|
|
print("Generating inference code for model: {}".format(_MODEL_PATH.value))
|
|
|
|
inference_generator.generate(output_dir, output_name,
|
|
graph.OpCodeTable([model]), graph.Graph(model))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(main)
|