我目前正在为对象识别任务构建一个(简单的顺序)全卷积网络。
设计架构本质上等于为每一层选择内核大小、步幅和单元数。这样做时,我必须跟踪我的图层的几个属性:
- 参数计数
- 每个内核的视野,输入/数据
- 在创建标签图像时我需要考虑的最后一层的边距
我编写了一个小脚本来计算不同 CNN 配置的这些数量。这不是很复杂,但是……
我的印象是每个使用 CNN 的人都必须这样做,所以应该有一个(图形)工具来解决这个问题。它也可以用来绘制精美的架构图。因为它适用于相同的输入数据。
有这样的工具吗?你如何为你的模型计算这些数量?