用于深度学习的微分几何计算机视觉
torchgeometr的Python项目详细描述
pytorch geometry包是用于PyTorch的几何计算机视觉库。
它由一组程序和可微模块组成,用于解决一般几何计算机视觉问题。在其核心,包使用pytorch作为其主要后端,既提高了效率,又利用了反向模式自动微分来定义和计算复杂函数的梯度。
安装
来自pip:
pip install torchgeometry
来源:
python setup.py install
来自使用pip的源代码:
pip install git+https://github.com/arraiyopensource/torchgeometry
快速使用
importtorchimporttorchgeometryastgmx_rad=tgm.pi*torch.rand(1,3,3)x_deg=tgm.rad2deg(x_rad)torch.allclose(x_rad,tgm.deg2rad(x_deg))# True
示例
运行我们的jupyter笔记本examples来学习使用库。
引用
如果您在研究相关文档中使用火炬几何,建议您引用海报。
@misc{Arraiy2018, author={E. Riba, M. Fathollahi, W. Chaney, E. Rublee and G. Bradski}title={torchgeometry: when PyTorch meets geometry}, booktitle={PyTorch Developer Conference}, year={2018}, url={https://drive.google.com/file/d/1xiao1Xj9WzjJ08YY_nYwsthE-wxfyfhG/view?usp=sharing}}
贡献
我们感谢所有的贡献。如果你打算贡献回错误修复,请这样做,没有任何进一步的讨论。如果您计划提供新功能、实用程序功能或扩展,请首先打开一个问题并与我们讨论该功能。请考虑阅读CONTRIBUTING注释。