Skip to main content

Logo

Build Status

This is the Framework for Easily Invertible Architectures (FrEIA).

  • Construct Invertible Neural Networks (INNs) from simple invertible building blocks.

  • Quickly construct complex invertible computation graphs and INN topologies.

  • Forward and inverse computation guaranteed to work automatically.

  • Most common invertible transforms and operations are provided.

  • Easily add your own invertible transforms.

Papers

Our following papers use FrEIA, with links to code given below.

“Generative Classifiers as a Basis for Trustworthy Image Classification” (CVPR 2021)

“Training Normalizing Flows with the Information Bottleneck for Competitive Generative Classification” (Neurips 2020)

“Disentanglement by Nonlinear ICA with General Incompressible-flow Networks (GIN)” (ICLR 2020)

“Guided Image Generation with Conditional Invertible Neural Networks” (2019)

“Analyzing inverse problems with invertible neural networks.” (ICLR 2019)

Installation

FrEIA has the following dependencies:

Package

Version

Python

>= 3.7

Pytorch

>= 1.0.0

Numpy

>= 1.15.0

Scipy

>= 1.5

Through pip

pip install git+https://github.com/VLL-HD/FrEIA.git

Manually

For development:

# first clone the repository
git clone https://github.com/VLL-HD/FrEIA.git
cd FrEIA
# install the dependencies
pip install -r requirements.txt
# install in development mode, so that changes don't require a reinstall
python setup.py develop

Documentation

The full manual can be found at https://vll-hd.github.io/FrEIA including

Metadata

Release files for FrEIA 0.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 FrEIA 0.2
File Size Uploaded
FrEIA-0.2.tar.gz 34.3 kB Details

Release files / FrEIA-0.2.tar.gz

Download URL FrEIA-0.2.tar.gz
Size 34.3 kB
Tags Source
SHA-256 checksum
How to use checksums
e9f1732b4a238b85b0ad3bc14b279a653a6063d14152028e164b989582026bdc
BLAKE2b-256 checksum
How to use checksums
d4765660f714a3a8c8df9c3301161b53fbdaa7a911ee0f660eaa65fa5a5b36f9
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.1 CPython/3.7.11

Release history Release notifications | RSS feed

This release

0.2 This release

1 release file

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