为了优化模型的CPU占用以及运行时间,使用TensorFlow提供的测试脚本benchmark_model
进行模型分析:
bazel-bin/tensorflow/tools/benchmark/benchmark_model \
--graph=/home/czczcz/test/mosaic512tv2Q.pb --input_layer="input" \
--input_layer_shape="1,299,299,3" --input_layer_type="float" \
--output_layer="output" --show_run_order=true --show_time=true \
--show_memory=true --show_summary=true --show_flops=true
得到的结果如下: