首页 > 解决方案 > 使用自定义数据集量化大小问题训练的 mobilenetv1

问题描述

我正在开发一个对象检测软件,基本上我在 Python 上使用带有 MobileNetV1 的 TensorFlow 对象检测 API,我已经用我自己的数据集训练了模型。

用我的数据集训练得到的 freeze_inference_graph.pb 文件就像 22 Mo。

我尝试通过量化将其转换为 TFLite,但它仍然像 21.2 Mo。

这两个尺寸都是 20+ Mo 正常吗?我从不同的来源了解到 MobileNet 量化模型大约是 5 Mo。这是因为我在自定义数据集上使用新对象对其进行了训练?而且,为什么量化它不会减小大小(最多小 4 倍)?

谢谢您的帮助

标签: tensorflowobject-detection-apiquantizationtensorflow-litemobilenet

解决方案


推荐阅读