fighting1996

Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

pytorch-doc-zh

Pytorch 中文文档

flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

awesome-semantic-segmentation-pytorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)

Self-Attention-GAN

Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)

senet.pytorch

PyTorch implementation of SENet

CBAM.PyTorch

Non-official implement of Paper：CBAM: Convolutional Block Attention Module

RCF

Richer Convolutional Features for Edge Detection

SENet-PyTorch

SKNet

try to implement the CVPR 2019 paper "Selective Kernel Networks" by PyTorch

EGNet

EGNet:Edge Guidance Network for Salient Object Detection (ICCV 2019)

cvpr2019_Pyramid-Feature-Attention-Network-for-Saliency-detection

code and model of Pyramid Feature Selective Network for Saliency detection

PiCANet-Implementation

Pytorch Implementation of PiCANet: Learning Pixel-wise Contextual Attention for Saliency Detection

DSS-pytorch

:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection

sal_eval_toolbox

evaluation toolbox for salient object detection

FPN_Pytorch

Base jwyang/fpn.pytorch, train FPN on Pascal VOC get 80.5 mAP

SE_DenseNet

This is a SE_DenseNet which contains a senet (Squeeze-and-Excitation Networks by Jie Hu, Li Shen, and Gang Sun) module, written in Pytorch, train, and eval codes have been released.

Spatial-Transformer-Networks_pytorch

Spatial Transformer Networks. Refer to daviddao/spatial-transformer-tensorflow.

plot-edge-pr-curves

The code and data to plot edge PR curves for many existing edge detectors

salience_object_detection

SalMetric

C++ code (with python wrapper) for measuring the F-measure and MAE scores for salient object detection

SKNet_pytorch

A reimplementation of the SKNet(Selective Kernel Networks) by pytorch

HRSOD

Implement-of-SCA-CNN

Implement-of-SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Caption

RFCN

This is a Pytorch implementation of RFCN[1] for saliency detection.

gradcam-pytorch

PyTorch Implement of Grad-CAM

PiCANet_Transfer_Learning_on_Neurons_PyTorch

PiCANet Transfer Learning on Neurons

PiCANet