ildoonet / Pytorch Randaugment
Licence: mit
Unofficial PyTorch Reimplementation of RandAugment.
Stars: ✭ 323
Programming Languages
python
139335 projects - #7 most used programming language
Labels
Projects that are alternatives of or similar to Pytorch Randaugment
Constrained attention filter
(ECCV 2020) Tensorflow implementation of A Generic Visualization Approach for Convolutional Neural Networks
Stars: ✭ 36 (-88.85%)
Mutual labels: classification, convolutional-neural-networks, imagenet
Cnn Svm
An Architecture Combining Convolutional Neural Network (CNN) and Linear Support Vector Machine (SVM) for Image Classification
Stars: ✭ 170 (-47.37%)
Mutual labels: classification, convolutional-neural-networks
Assembled Cnn
Tensorflow implementation of "Compounding the Performance Improvements of Assembled Techniques in a Convolutional Neural Network"
Stars: ✭ 319 (-1.24%)
Mutual labels: convolutional-neural-networks, imagenet
Mobilenet V2
A Complete and Simple Implementation of MobileNet-V2 in PyTorch
Stars: ✭ 206 (-36.22%)
Mutual labels: classification, convolutional-neural-networks
Practical Machine Learning With Python
Master the essential skills needed to recognize and solve complex real-world problems with Machine Learning and Deep Learning by leveraging the highly popular Python Machine Learning Eco-system.
Stars: ✭ 1,868 (+478.33%)
Mutual labels: classification, convolutional-neural-networks
Efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras.
Stars: ✭ 1,920 (+494.43%)
Mutual labels: classification, imagenet
Cnn 3d Images Tensorflow
3D image classification using CNN (Convolutional Neural Network)
Stars: ✭ 199 (-38.39%)
Mutual labels: classification, convolutional-neural-networks
Pytorch Imagenet Cifar Coco Voc Training
Training examples and results for ImageNet(ILSVRC2012)/CIFAR100/COCO2017/VOC2007+VOC2012 datasets.Image Classification/Object Detection.Include ResNet/EfficientNet/VovNet/DarkNet/RegNet/RetinaNet/FCOS/CenterNet/YOLOv3.
Stars: ✭ 130 (-59.75%)
Mutual labels: classification, imagenet
Mead Baseline
Deep-Learning Model Exploration and Development for NLP
Stars: ✭ 238 (-26.32%)
Mutual labels: classification, convolutional-neural-networks
simpleAICV-pytorch-ImageNet-COCO-training
SimpleAICV:pytorch training example on ImageNet(ILSVRC2012)/COCO2017/VOC2007+2012 datasets.Include ResNet/DarkNet/RetinaNet/FCOS/CenterNet/TTFNet/YOLOv3/YOLOv4/YOLOv5/YOLOX.
Stars: ✭ 276 (-14.55%)
Mutual labels: classification, imagenet
classification
Catalyst.Classification
Stars: ✭ 35 (-89.16%)
Mutual labels: classification, augmentation
Lsoftmax Pytorch
The Pytorch Implementation of L-Softmax
Stars: ✭ 133 (-58.82%)
Mutual labels: classification, convolutional-neural-networks
Vision4j Collection
Collection of computer vision models, ready to be included in a JVM project
Stars: ✭ 132 (-59.13%)
Mutual labels: classification, imagenet
Shufflenet V2 Tensorflow
A lightweight convolutional neural network
Stars: ✭ 145 (-55.11%)
Mutual labels: classification, imagenet
Simpsonrecognition
Detect and recognize The Simpsons characters using Keras and Faster R-CNN
Stars: ✭ 131 (-59.44%)
Mutual labels: classification, convolutional-neural-networks
Imgclsmob
Sandbox for training deep learning networks
Stars: ✭ 2,405 (+644.58%)
Mutual labels: classification, imagenet
CvT
This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.
Stars: ✭ 262 (-18.89%)
Mutual labels: classification, imagenet
Handwritten Digit Recognition Using Deep Learning
Handwritten Digit Recognition using Machine Learning and Deep Learning
Stars: ✭ 127 (-60.68%)
Mutual labels: classification, convolutional-neural-networks
Regnet
Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"
Stars: ✭ 129 (-60.06%)
Mutual labels: classification, imagenet
Transfer Learning Suite
Transfer Learning Suite in Keras. Perform transfer learning using any built-in Keras image classification model easily!
Stars: ✭ 212 (-34.37%)
Mutual labels: classification, convolutional-neural-networks
pytorch-randaugment
Unofficial PyTorch Reimplementation of RandAugment. Most of codes are from Fast AutoAugment.
Introduction
Models can be trained with RandAugment for the dataset of interest with no need for a separate proxy task. By only tuning two hyperparameters(N, M), you can achieve competitive performances as AutoAugments.
Install
$ pip install git+https://github.com/ildoonet/pytorch-randaugment
Usage
from torchvision.transforms import transforms
from RandAugment import RandAugment
transform_train = transforms.Compose([
transforms.RandomCrop(32, padding=4),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize(_CIFAR_MEAN, _CIFAR_STD),
])
# Add RandAugment with N, M(hyperparameter)
transform_train.transforms.insert(0, RandAugment(N, M))
Experiment
We use same hyperparameters as the paper mentioned. We observed similar results as reported.
You can run an experiment with,
$ python RandAugment/train.py -c confs/wresnet28x10_cifar10_b256.yaml --save cifar10_wres28x10.pth
CIFAR-10 Classification
Model | Paper's Result | Ours |
---|---|---|
Wide-ResNet 28x10 | 97.3 | 97.4 |
Shake26 2x96d | 98.0 | 98.1 |
Pyramid272 | 98.5 |
CIFAR-100 Classification
Model | Paper's Result | Ours |
---|---|---|
Wide-ResNet 28x10 | 83.3 | 83.3 |
SVHN Classification
Model | Paper's Result | Ours |
---|---|---|
Wide-ResNet 28x10 | 98.9 | 98.8 |
ImageNet Classification
I have experienced some difficulties while reproducing paper's result.
Issue : https://github.com/ildoonet/pytorch-randaugment/issues/9
Model | Paper's Result | Ours |
---|---|---|
ResNet-50 | 77.6 / 92.8 | TODO |
EfficientNet-B5 | 83.2 / 96.7 | TODO |
EfficientNet-B7 | 84.4 / 97.1 | TODO |
References
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].