图文件的大小(.pb)从Tensorflow是太大的android使用,如何减少它



我尝试使用tensorflow构建一个实现Yolo对象检测的网络,我希望它可以在Android上使用。在构建结构之后,我使用tf.train。Write_graph获取图形文件,并希望在android demo中替换原始文件。但是pb文件太大(1.1G),在Android上无法使用。那么,我怎样才能减小尺寸呢?

我建议你首先尝试量化你的图,因为你只需要一个官方的TensorFlow脚本。这是Pete Warden的一个很棒的教程:https://petewarden.com/2016/05/03/how-to-quantize-neural-networks-with-tensorflow/

理论上,如果你使用32位浮点数,你的模型最终会小4倍(~250Mb),因为图中的值将被转换为8位整数(推断它对性能没有显著影响)。注意,这在压缩协议缓冲区文件时起作用。

相关内容