keras-non-local-nets
keras-non-local-nets项目提供了Keras实现的非局部神经网络块。支持Gaussian、Embedded Gaussian和Dot等多种实例化方式,并通过可变屏蔽计算模式优化性能。项目包含使用模板和示例代码,便于集成到现有神经网络中。同时支持函数式API和Sequential API,适应不同的网络构建需求。