3D Unet

PyTorch implementation 3D U-Net and its variants.

The code allows for training the U-Net for both: semantic segmentation (binary and multi-class) and regression problems (e.g. …

Keywords: Unet, U-net, 3D U-net, semantic segmentation, computer vision, PyTorch, de-noising, deep learning, neural networks

Adversarial Latent Autoencoders

Reference implementation for the paper Adversarial Latent Autoencoders by S. Pidhorskyi, D. A. Adjeroh, and G. Doretto, CVPR 2020.

Keywords: ALAE, StyleALAE, adversarial latent autoencoders, computer vision, image processing, deep learning, autoencoder, unsupervised learning

AnimeGAN

This is the Open source of the paper "AnimeGAN: a novel lightweight GAN for photo animation", which uses the GAN framwork to transform real-world photos …

Keywords: AnimeGAN, Tensorflow, generative adversarial networks, deep learning, computer vision, anime, photo animation

Asteroid

Asteroid is a Pytorch-based audio source separation toolkit that enables fast experimentation on common datasets. It comes with a source code that supports a large …

Keywords: Asteroid, Pytorch, audio source separation, toolkit, audio, sound processing, ConvTasnet, Tasnet, Deep clustering, DualPathRNN, Chimera++, Wavesplit

Attention-based Deep Multiple Instance Learning

The official implementation of a method for multiple instance learning using deep neural networks and attention mechanism.

Keywords: multiple instance learning, deep learning, attention, PyTorch, weak classification

Attri2Vec

The reference implementation for the attri2vec algorithm.

Keywords: Attri2vec, representation learning, graph machine learning, node classification, link prediction

Backprop

Backprop is a Python library for fine-tuning and deploying machine learning models. It includes pre-trained models for natural language processing, e.g., question answering, text summarisation, …

Keywords: neural networks, pre-trained models, fine-tuning, library, natural language processing, computer vision, Python, PyTorch

CartoonGAN Pytorch

Pytorch implementation of the paper CartoonGAN: Generative Adversarial Networks for Photo Cartoonization by Y. Chen, Y. Lai, and Y. Liu.

Keywords: CartoonGAN, Pytorch, generative adversarial networks, deep learning, computer vision, cartoonization

CartoonGAN Tensorflow

Tensorflow implementation of the paper CartoonGAN: Generative Adversarial Networks for Photo Cartoonization by Y. Chen, Y. Lai, and Y. Liu.

Keywords: CartoonGAN, Tensorflow, generative adversarial networks, deep learning, computer vision, cartoonization

Cluster-GCN Tensorflow

This repository contains the official TensorFlow implementation of the paper Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks by W. Chiang, …

Keywords: Cluster-GCN, Tensorflow, graph convolutional networks, geometric deep learning, graph machine learning, GCN