文件名称：pytorch-gdn:广义除法归一化非线性层的PyTorch实现
文件大小：4KB
文件格式：ZIP
更新时间：2024-05-21 19:24:02
Python
PyTorch GDN 广义除法归一化层 该实用程序基于以下文章提供了GDN非线性的PyTorch实现： “使用广义归一化变换对图像进行密度建模” 约翰内斯·巴拉（JohannesBallé），瓦莱罗·拉帕拉（Valero Laparra），埃罗·P·西蒙切利（Eero P.Simoncelli） “端到端优化图像压缩” 约翰内斯·巴拉（JohannesBallé），瓦莱罗·拉帕拉（Valero Laparra），埃罗·P·西蒙切利（Eero P.Simoncelli） 该实现基于contrib包（ ）上可用的Tensorflow实现 用法 GDN层可用作PyTorch中的常规非线性，但必须在应用程序和将使用该设备的割炬设备上用通道数实例化。 GDN层支持4-d输入（图像批处理）或5-d输入（视频批处理）。 通过展开序列尺寸来处理5维输入。 device = torch.de
【文件预览】：
pytorch-gdn-master
----.gitignore(1KB)
----README.md(1KB)
----pytorch_gdn()
--------__init__.py(3KB)
----LICENSE(1KB)
----setup.py(399B)