文件名称:CBAM_MNIST.py
文件大小:12KB
文件格式:PY
更新时间:2023-06-05 12:22:50
tensorflow 注意力机制
构建一个四层神经网络识别手写体数据集MNIST,然后将注意力模块CBAM插入到网络的第一层之后,查看注意力模块的性能。可以改变CBAM模块插入的位置,做到任意插入。
文件名称:CBAM_MNIST.py
文件大小:12KB
文件格式:PY
更新时间:2023-06-05 12:22:50
tensorflow 注意力机制
构建一个四层神经网络识别手写体数据集MNIST,然后将注意力模块CBAM插入到网络的第一层之后,查看注意力模块的性能。可以改变CBAM模块插入的位置,做到任意插入。