xyaozi

eng-practices

Google's Engineering Practices documentation

generative-models

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

torch2trt

An easy to use PyTorch to TensorRT converter

pytorch-beginner

pytorch tutorial for beginners

pytorch-ssim

pytorch structural similarity (SSIM) loss

pytorch_classification

利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码

GCNet

GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond

pytorch-msssim

Fast and differentiable MS-SSIM and SSIM for pytorch.

CenterMask

[CVPR 2020] CenterMask : Real-Time Anchor-Free Instance Segmentation

deeplabv3

PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.

deeplabv3plus-pytorch

Here is a pytorch implementation of deeplabv3+ supporting ResNet(79.155%) and Xception(79.945%). Multi-scale & flip test and COCO dataset interface has been finished.

kmeans-anchor-boxes

k-means clustering with the Intersection over Union (IoU) metric as described in the YOLO9000 paper

ADGAN

The Implementation of paper "Controllable Person Image Synthesis with Attribute-Decomposed GAN" CVPR 2020 (Oral); Pose and Appearance Attributes Transfer;

pytorch-vae

A Variational Autoencoder (VAE) implemented in PyTorch

DetNet_pytorch

An implementation of DetNet: A Backbone network for Object Detection.

vovnet-detectron2

[CVPR 2020] VoVNet backbone networks for detectron2

pytorch-vae

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

KD_methods_with_TF

Knowledge distillation methods implemented with Tensorflow (now there are 11 (+1) methods, and will be added more.)

Anomaly-Detection-PatchSVDD-PyTorch

RAPiD

RAPiD: Rotation-Aware People Detection in Overhead Fisheye Images (CVPR 2020 Workshops)

dcgan_vae_pytorch

dcgan combined with vae in pytorch!

pytorch-mnist-VAE

pytorch-AutoEncoders

Autoencoders in PyTorch

pytorch-AE

Autoencoders in PyTorch

pytorch-autoencoders

Implementation of autoencoders in PyTorch

OCGAN

SSD_mobilenetv2-with-Focal-loss

this repo is forked from https://github.com/amdegroot/ssd.pytorch. Implemented by pytorch

mobilenetv2_deeplabv3_pytorch

Try to implement deeplab v3+ on pytorch according to offical demo.

cnn_autoencoder_mnist

CNN auto-encoder on mnist.

AutoEncoders

Variational autoencoder, denoising autoencoder and other variations of autoencoders implementation in keras