Skip to main content

Discrete Cosine Transform (DCT) for pytorch

Project description

DCT (Discrete Cosine Transform) for pytorch

Build Status codecov PyPI version PyPI version PyPI status GitHub license

This library implements DCT in terms of the built-in FFT operations in pytorch so that back propagation works through it, on both CPU and GPU. For more information on DCT and the algorithms used here, see Wikipedia and the paper by J. Makhoul. This StackExchange article might also be helpful.

The following are currently implemented:

  • 1-D DCT-I and its inverse (which is a scaled DCT-I)
  • 1-D DCT-II and its inverse (which is a scaled DCT-III)
  • 2-D DCT-II and its inverse (which is a scaled DCT-III)
  • 3-D DCT-II and its inverse (which is a scaled DCT-III)

Install

pip install torch-dct

Requires torch>=0.4.1 (lower versions are probably OK but I haven't tested them).

You can run test by getting the source and run pytest. To run the test you also need scipy installed.

Usage

import torch
import torch_dct as dct

x = torch.randn(200)
X = dct.dct(x)   # DCT-II done through the last dimension
y = dct.idct(X)  # scaled DCT-III done through the last dimension
assert (torch.abs(x - y)).sum() < 1e-10  # x == y within numerical tolerance

dct.dct1 and dct.idct1 are for DCT-I and its inverse. The usage is the same.

Just replace dct and idct by dct_2d, dct_3d, idct_2d, idct_3d, etc to get the multidimensional versions.

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

torch-dct-0.1.6.tar.gz (4.8 kB view details)

Uploaded Source

Built Distribution

torch_dct-0.1.6-py3-none-any.whl (5.1 kB view details)

Uploaded Python 3

File details

Details for the file torch-dct-0.1.6.tar.gz.

File metadata

  • Download URL: torch-dct-0.1.6.tar.gz
  • Upload date:
  • Size: 4.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.1 CPython/3.10.6

File hashes

Hashes for torch-dct-0.1.6.tar.gz
Algorithm Hash digest
SHA256 87923c4b8430206e37f1ccaa80386b8600d4f65921741399cb99afdb61a3731a
MD5 d8ffcf612a6dee40796ebf3d9bbe6fff
BLAKE2b-256 2c12518883ae7a1f0bd1039ddbb0ab3934a8cc8b1d609b47b1b7645eda59c72b

See more details on using hashes here.

File details

Details for the file torch_dct-0.1.6-py3-none-any.whl.

File metadata

  • Download URL: torch_dct-0.1.6-py3-none-any.whl
  • Upload date:
  • Size: 5.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.1 CPython/3.10.6

File hashes

Hashes for torch_dct-0.1.6-py3-none-any.whl
Algorithm Hash digest
SHA256 6ab1a064e7f138b649148c01e06cf0f0d354cdb3418ef0e04c81576ce9ba656b
MD5 ae7001b5b663abdfd0e24bcc9a2bf16b
BLAKE2b-256 87883eef09f85bc6f22d78d820d8af91e3823a448fbee41ef53a35087297ed63

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