用于设计 CNN 架构的工具

数据挖掘 工具 美国有线电视新闻网
2022-03-10 04:47:10

我目前正在为对象识别任务构建一个(简单的顺序)全卷积网络。

设计架构本质上等于为每一层选择内核大小、步幅和单元数。这样做时,我必须跟踪我的图层的几个属性:

  • 参数计数
  • 每个内核的视野,输入/数据
  • 在创建标签图像时我需要考虑的最后一层的边距

我编写了一个小脚本来计算不同 CNN 配置的这些数量。这不是很复杂,但是……

我的印象是每个使用 CNN 的人都必须这样做,所以应该有一个(图形)工具来解决这个问题。它也可以用来绘制精美的架构图。因为它适用于相同的输入数据。

有这样的工具吗?你如何为你的模型计算这些数量?

1个回答

鉴于参数的数量和每个参数的选项数量,图形工具(又名图形用户界面或 GUI)对于 CNN 来说不是很实用。大多数从业者以文本形式存储和操作参数。

TensorFlow 有两个图形项目。Playground是一个浏览器内网络,可以在样本数据集上修改和运行。TensorBoard可视化计算图并帮助调试模型。