我尝试构建一个使用 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 位 float ，您的模型将最终缩小约 4 倍(约 250Mb)，因为图中的值将转换为 8 位整数(对于推理，它对性能没有显着影响).请注意，当您压缩 Protocol Buffer 文件时，这会起作用。
关于android - 来自 Tensorflow 的图形文件(.pb)的大小对于 android 使用来说太大了，如何减少它？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/39441107/