senet.pytorch

SE-Nets

An implementation of Squeeze-and-Excitation Networks (SE-Nets) for deep learning image classification tasks

PyTorch implementation of SENet

GitHub

2k stars
16 watching
442 forks
Language: Python
last commit: over 3 years ago
Linked from 1 awesome list

pytorchsenet

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
cadene/pretrained-models.pytorch Provides pre-trained PyTorch models for various computer vision tasks and datasets. 9,036
hujie-frank/senet An implementation of the Squeeze-and-Excitation Networks architecture in Caffe 3,394
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 83,959
prlz77/resnext.pytorch Reproduces ResNet-V3 with PyTorch for computer vision tasks 508
taki0112/senet-tensorflow A TensorFlow implementation of Squeeze and Excitation Networks for image classification tasks 756
rosinality/stylegan2-pytorch An implementation of StyleGAN 2 in PyTorch for image generation and analysis. 2,741
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,147
timesler/facenet-pytorch This project provides pre-trained PyTorch models for face detection and recognition tasks. 4,550
meetps/pytorch-semseg Implementations of popular semantic segmentation architectures using PyTorch 3,393
qubvel-org/segmentation_models.pytorch A PyTorch library for building and training neural networks for image segmentation tasks. 9,696
cszn/kair Image restoration toolbox with training and testing codes for various deep learning-based methods 2,957
stvir/pysot A software system designed to support research in visual tracking using deep learning algorithms 4,438
ericsun99/shufflenet-v2-pytorch A PyTorch implementation of a ShuffleNet-v2 neural network architecture for image classification. 430
amdegroot/ssd.pytorch An implementation of a deep learning-based object detection system in PyTorch. 5,146
diego999/pygat An implementation of the Graph Attention Network model using PyTorch. 2,914