beauty-net

Image classifier

Provides a basic framework for training deep learning models on image classification tasks using PyTorch

A simple, flexible, and extensible template for PyTorch. It's beautiful.

GitHub

187 stars
7 watching
11 forks
Language: Python
last commit: about 1 year ago
Linked from 2 awesome lists

cnncomputer-visiondeep-learningimage-classificationpytorchpytorch-template

Backlinks from these awesome lists:

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
lucasjinreal/pytorch_image_classifier An image classification system built using PyTorch, capable of fine-tuning on various models and datasets. 113
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
tengshaofeng/residualattentionnetwork-pytorch Implementing a deep learning framework for image classification using Residual Attention Network architecture 680
bmsookim/wide-resnet.pytorch PyTorch implementation of wide residual networks for image classification tasks on CIFAR-10 and CIFAR-100 datasets 461
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,692
blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorch An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts 65
chenxi116/pnasnet.tf An implementation of PNASNet-5 architecture in TensorFlow for image classification on ImageNet. 102
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
openseg-group/openseg.pytorch Provides a PyTorch implementation of several computer vision tasks including object detection, segmentation and parsing. 1,190
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
hszhao/pspnet A PyTorch implementation of a deep learning model for semantic image segmentation 1,593
microsoft/cvt An implementation of a new neural network architecture that combines the strengths of convolutional and transformer designs to improve performance on image classification tasks. 555
benedekrozemberczki/seal-ci A PyTorch implementation of a semi-supervised graph classification model that learns hierarchical representations from labeled and unlabeled graph data. 208