减小张量流模型文件的大小

Posted 2023-03-13

技术标签:

【中文标题】减小张量流模型文件的大小【英文标题】：Reducing the size of a tensorflow model file 【发布时间】：2021-09-07 16:14:12 【问题描述】：

我有一个 TensorFlow 模型，它基本上将材料分为 X 射线和非 X 射线。

我的问题是 TensorFlow 模型文件（对于该模型）高达 123mb 大，我需要以某种方式减小它的大小。

我在 Stack Overflow 上看到了 answer 这里有我需要做的确切代码，但仅限于 Tensorflow 1.0。

因此，如果这里有人能给我更新的 TensorFlow 2.0 代码，我将不胜感激。

Python 版本：Python 3.8.7

【问题讨论】：

【参考方案1】：

密集层在存储模型中需要大尺寸。使用更多的卷积层和更少的密集层。在 model.summary() 中，你可以看到每一层存储了多少参数

【讨论】：

以上是关于减小张量流模型文件的大小的主要内容，如果未能解决你的问题，请参考以下文章