Deep_Learning_model_profiler 源码

Model_profier 一个工具可以帮助您计算DL模型的神经元编号,参数和操作数 图书馆 Keras 2.4.0 Tensorflow 2.2.0 当前支持的图层: 转换2d 线性的 批量标准化2D 最大池化2D 平均池化2D 稠密 测试 您可以使用预先训练的模型测试脚本并保存其统计信息 您可以在标准CNN型号VGG16 / ResNet50 /上进行测试。 在models.py建立自己的CNN模型 输出量 所有模型信息都将记录在“ statistics_files”文件夹中的“ * .txt”中。 FLOPs计算 为了计算浮点运算(FLOP)的数量,我们假设将卷积实现为滑