jiweibo / ImageNet
This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)
☆420Updated 6 years ago
Alternatives and similar repositories for ImageNet:
Users that are interested in ImageNet are comparing it to the libraries listed below
- Classification with PyTorch.☆1,724Updated 10 months ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,272Updated 10 months ago
- This is the PyTorch implementation of VGG network trained on CIFAR10 dataset☆351Updated 5 years ago
- 3.41% and 17.11% error on CIFAR-10 and CIFAR-100☆332Updated 6 years ago
- experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN a…☆727Updated 6 years ago
- pytorch and cifar10☆216Updated 4 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆746Updated 4 years ago
- PyTorch implementation of SENet☆2,319Updated 4 years ago
- VGG16 Net implementation from PyTorch Examples scripts for ImageNet dataset☆86Updated 4 years ago
- A PyTorch implementation of VGG16. This could be considered as a variant of the original VGG16 since BN layers are added after each conv…☆185Updated 5 years ago
- Network Slimming (Pytorch) (ICCV 2017)☆915Updated 4 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,403Updated 3 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,320Updated 3 years ago
- SimpleAICV:pytorch training and testing examples.☆432Updated last month
- Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeN…☆2,664Updated 2 years ago
- Reproduces ResNet-V3 with pytorch☆514Updated 4 years ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆863Updated 4 years ago
- 72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models☆698Updated 4 years ago
- A memory-efficient implementation of DenseNets☆1,533Updated last year
- Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"☆2,136Updated 2 years ago
- ☆407Updated 7 years ago
- ☆218Updated 7 years ago
- Wide Residual Networks (WideResNets) in PyTorch☆338Updated 3 years ago
- try to implement the CVPR 2019 paper "Selective Kernel Networks" by PyTorch☆375Updated 5 years ago
- PyTorch implementation of CNNs for CIFAR benchmark☆705Updated 4 years ago
- Implementation of Non-local Block.☆1,572Updated 2 years ago
- A PyTorch Implementation for Densely Connected Convolutional Networks (DenseNets)☆470Updated 7 years ago
- mixup: Beyond Empirical Risk Minimization☆1,178Updated 3 years ago
- Non-official implement of Paper：CBAM: Convolutional Block Attention Module☆1,399Updated last year
- PyTorch implementation of AlexNet☆127Updated 7 years ago