Skip to main content
https://travis-ci.com/arraiyopensource/torchgeometry.svg?branch=master https://codecov.io/github/arraiyopensource/torchgeometry/branch/master/graph/badge.svg https://badge.fury.io/py/torchgeometry.svg Documentation Status

The PyTorch Geometry package is a geometric computer vision library for PyTorch.

It consists of a set of routines and differentiable modules to solve generic geometry computer vision problems. At its core, the package uses PyTorch as its main backend both for efficiency and to take advantage of the reverse-mode auto-differentiation to define and compute the gradient of complex functions.

Installation

From pip:

pip install torchgeometry

From source:

python setup.py install

From source using pip:

pip install git+https://github.com/arraiyopensource/torchgeometry

Quick Usage

import torch
import torchgeometry as tgm

x_rad = tgm.pi * torch.rand(1, 3, 3)
x_deg = tgm.rad2deg(x_rad)

torch.allclose(x_rad, tgm.deg2rad(x_deg))  # True

Examples

Run our Jupyter notebooks examples to learn to use the library.

Cite

If you are using torchgeometry in your research-related documents, it is recommended that you cite the poster.

@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

We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so without any further discussion. If you plan to contribute new features, utility functions or extensions, please first open an issue and discuss the feature with us. Please, consider to read the CONTRIBUTING notes.

Release files for torchgeometry 0.1.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for torchgeometry 0.1.2
File Size Uploaded
torchgeometry-0.1.2.tar.gz 44.3 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for torchgeometry 0.1.2
File Interpreter ABI Platform
torchgeometry-0.1.2-py2.py3-none-any.whl Python 3, Python 2 none any Details

Total release size: 87.0 kB

Release files / torchgeometry-0.1.2.tar.gz

Download URL torchgeometry-0.1.2.tar.gz
Size 44.3 kB
Tags Source
SHA-256 checksum
How to use checksums
5d9129e709baaeece5298d9d27a3a3961ac0659264846392138c785520bdb872
BLAKE2b-256 checksum
How to use checksums
3ebb5a7c43067349a2e85a3ccb249e4a2ff6518358e40be1a3062f94f34ca8e9
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/1.13.0 pkginfo/1.5.0 requests/2.21.0 setuptools/40.8.0 requests-toolbelt/0.8.0 tqdm/4.29.0 CPython/3.7.1

Release files / torchgeometry-0.1.2-py2.py3-none-any.whl

Download URL torchgeometry-0.1.2-py2.py3-none-any.whl
Size 42.7 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
f4b08b19359581f76aa93206fac4b439fbe784be2f072fca7d7e720bf0e28ce0
BLAKE2b-256 checksum
How to use checksums
a6d63f6820c0589bc3876080c59b58a3bad11af746a7b46f364b1cde7972bd72
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/1.13.0 pkginfo/1.5.0 requests/2.21.0 setuptools/40.8.0 requests-toolbelt/0.8.0 tqdm/4.29.0 CPython/3.7.1

Release history Release notifications | RSS feed

This release

0.1.2 This release

2 release files

0.1.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page