《表1 注意力记忆网络结构》
注意力记忆网络使用Keras框架[11]开发,TensorFlow[12]作为后端。将注意力记忆模块中记忆模块的BiGRU神经元个数设为128,其中神经元的个数是64,双向GRU神经元个数翻倍。注意力记忆模块和卷积残差模块的网络结构如表1所示。该网络使用SGD随机梯度下降[13]的方法来完成优化训练,其学习率设为0.1,学习参数momentum设为0.9。
图表编号 | XD00133707900 严禁用于非法目的 |
---|---|
绘制时间 | 2020.04.15 |
作者 | 盛龙帅、李策、李欣 |
绘制单位 | 江苏省社会安全图像与视频理解重点实验室(南京理工大学)、中国矿业大学(北京)机电与信息工程学院、江苏省社会安全图像与视频理解重点实验室(南京理工大学)、中国矿业大学(北京)机电与信息工程学院、中国矿业大学(北京)机电与信息工程学院 |
更多格式 | 高清、无水印(增值服务) |