Skip to main content

differential geometric computer vision for deep learning

Project description

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

torchgeometry-0.1.2.tar.gz (44.3 kB view details)

Uploaded Source

Built Distribution

torchgeometry-0.1.2-py2.py3-none-any.whl (42.7 kB view details)

Uploaded Python 2 Python 3

File details

Details for the file torchgeometry-0.1.2.tar.gz.

File metadata

  • Download URL: torchgeometry-0.1.2.tar.gz
  • Upload date:
  • Size: 44.3 kB
  • Tags: Source
  • Uploaded using 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

File hashes

Hashes for torchgeometry-0.1.2.tar.gz
Algorithm Hash digest
SHA256 5d9129e709baaeece5298d9d27a3a3961ac0659264846392138c785520bdb872
MD5 2069b07a2e6da2651426af72221e39b1
BLAKE2b-256 3ebb5a7c43067349a2e85a3ccb249e4a2ff6518358e40be1a3062f94f34ca8e9

See more details on using hashes here.

File details

Details for the file torchgeometry-0.1.2-py2.py3-none-any.whl.

File metadata

  • Download URL: torchgeometry-0.1.2-py2.py3-none-any.whl
  • Upload date:
  • Size: 42.7 kB
  • Tags: Python 2, Python 3
  • Uploaded using 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

File hashes

Hashes for torchgeometry-0.1.2-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 f4b08b19359581f76aa93206fac4b439fbe784be2f072fca7d7e720bf0e28ce0
MD5 ab17bf1d264cc7296b881d94b00b0222
BLAKE2b-256 a6d63f6820c0589bc3876080c59b58a3bad11af746a7b46f364b1cde7972bd72

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page