TensorFlow Tutorial and Examples for Beginners (support TF v1 & v2)

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"

Image augmentation library in Python for machine learning.

Siamese and triplet networks with online pair/triplet mining in PyTorch

Repository for the Explainable Deep One-Class Classification paper

Repository for the Deep One-Class Classification ICML 2018 paper

pytorch knn [cuda version]

GPU Accelerated t-SNE for CUDA with Python bindings

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms

PyTorch Implementation of Deep Feature Consistent Variational Autoencoder.

PyTorch implementations of Generative Adversarial Networks.

A CNN Variational Autoencoder (CNN-VAE) implemented in PyTorch

A high performance impermentation of Unsupervised Image Segmentation by Backpropagation - Asako Kanezaki

Datasets for industrial surface-inspection

AutoEncoder vs Metric Learning for Anomaly Detection

K-Nearest Neighbor in Pytorch

Pytorch Implementation of variational auto-encoder for MNIST

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Invariant Information Clustering for Unsupervised Image Classification and Segmentation

Invariant Information Clustering for Unsupervised Image Classification and Segmentation

Invariant Information Clustering for Unsupervised Image Classification and Segmentation

Convolutional Autoencoder with SetNet in PyTorch

Fast and differentiable MS-SSIM and SSIM for pytorch.

Saliency detection with a convolutional autoencoder including an edge contrast penalty term to the loss to enforce sharp edges

pytorch structural similarity (SSIM) loss