风格迁移模型性能测试

        为了优化模型的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

        得到的结果如下:


性能测试.png

你可能感兴趣的:(风格迁移模型性能测试)