pytorch-classification
Classification framework
An open-source project providing a unified interface for training and evaluating classification models using PyTorch.
Classification with PyTorch.
2k stars
30 watching
563 forks
Language: Python
last commit: 5 months ago cifar10cifar100classificationdensenetimagenetpreresnetpytorchresnetresnextwide-residual-networkswrn
Related projects:
| Repository | Description | Stars |
|---|---|---|
 felixgwu/img_classification_pk_pytorch | A PyTorch project for comparing image classification models and facilitating quick experiment setup | 365 |
 facebookresearch/pycls | A flexible PyTorch image classification framework for rapid research exploration and model evaluation. | 2,138 |
 achaiah/pywick | A PyTorch-based neural network training framework with advanced features and utilities | 398 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
 cmsflash/beauty-net | Provides a basic framework for training deep learning models on image classification tasks using PyTorch | 187 |
| bearpaw/pytorch-pose | Provides tools and infrastructure for training, testing, and evaluating 2D human pose estimation models | 1,105 |
 tengshaofeng/residualattentionnetwork-pytorch | Implementing a deep learning framework for image classification using Residual Attention Network architecture | 680 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,041 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 569 |
 bmsookim/wide-resnet.pytorch | PyTorch implementation of wide residual networks for image classification tasks on CIFAR-10 and CIFAR-100 datasets | 461 |
 erotemic/netharn | A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. | 39 |
 lucasjinreal/pytorch_image_classifier | An image classification system built using PyTorch, capable of fine-tuning on various models and datasets. | 113 |
 pyg-team/pytorch-frame | A deep learning framework for handling heterogeneous tabular data with diverse column types | 543 |
 ruotianluo/imagecaptioning.pytorch | A Python-based framework for training and testing image captioning models using PyTorch. | 1,451 |