peter31726

deep-high-resolution-net.pytorch

The project is an official implementation of our CVPR2019 paper "Deep High-Resolution Representation Learning for Human Pose Estimation"

OctaveConv_pytorch

Pytorch implementation of newly added convolution

SODsurvey

Salient Object Detection in the Deep Learning Era: An In-Depth Survey

mobilenetv3

mobilenetv3 with pytorch，provide pre-train model

drn

Dilated Residual Networks

BASNet

Code for CVPR 2019 paper. BASNet: Boundary-Aware Salient Object Detection

saliency-2019-SalBCE

SalGAN implementation with pytorch trained on [DHF1K, LEDOV, SALICON] using [BCE, Adversarial] loss.

SOD-CNNs-based-code-summary-

The summary of code and paper for salient object detection with deep learning.

cupy

NumPy & SciPy for GPU

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

salgan

SalGAN: Visual Saliency Prediction with Generative Adversarial Networks

salgan

Switchable-Normalization

Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779

pytorch-lr-finder

A learning rate range test implementation in PyTorch

onecyclelr

One cycle policy learning rate scheduler in PyTorch

senet.pytorch

PyTorch implementation of SENet

attentive-gan-derainnet

Unofficial tensorflow implemention of "Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018) " model https://maybeshewill-cv.github.io/attentive-gan-derainnet/

DeRaindrop

Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018)

SqueezeNet

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters

tensorpack

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

ShuffleNet-Series

PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

mobilenetv2.pytorch

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

pytorch-mobilenet-v2

A PyTorch implementation of MobileNet V2 architecture and pretrained model.

pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

examples

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

CycleGAN-tensorflow

Tensorflow implementation for learning an image-to-image translation without input-output pairs. https://arxiv.org/pdf/1703.10593.pdf

CycleGAN-TensorFlow

An implementation of CycleGan using TensorFlow

MSDNet

Multi-Scale Dense Networks for Resource Efficient Image Classification （ICLR 2018 Oral）

MSDNet-PyTorch

MSDNet