← all repositories

hysts/pytorch_image_classification

PyTorch implementations of deep learning models for image classification on CIFAR, MNIST, and ImageNet datasets.

1.4k stars Python Computer VisionML Frameworks
pytorch_image_classification
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of numerous deep learning architectures for image classification tasks. It includes models such as ResNet, DenseNet, PyramidNet, SENet, and various augmentation techniques like CutMix, Mixup, and Random Erasing. The code supports training on standard benchmark datasets including CIFAR-10, CIFAR-100, MNIST, FashionMNIST, and ImageNet with configurable settings via YAML files.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.