pytorch_image_classifier

Image classifier

An image classification system built using PyTorch, capable of fine-tuning on various models and datasets.

Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.

GitHub

113 stars
5 watching
48 forks
Language: Python
last commit: about 4 years ago

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
cmsflash/beauty-net Provides a basic framework for training deep learning models on image classification tasks using PyTorch 187
bmsookim/wide-resnet.pytorch PyTorch implementation of wide residual networks for image classification tasks on CIFAR-10 and CIFAR-100 datasets 461
ibm/max-inception-resnet-v2 An image classification model using a third-generation deep residual network. 28
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
xiayandi/pytorch_text_classification An implementation of convolutional neural networks for text classification using PyTorch 66
tengshaofeng/residualattentionnetwork-pytorch Implementing a deep learning framework for image classification using Residual Attention Network architecture 680
ruotianluo/imagecaptioning.pytorch A Python-based framework for training and testing image captioning models using PyTorch. 1,451
hrnet/hrnet-image-classification This project provides a Python implementation of high-resolution networks for image classification on ImageNet. 968
blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorch An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts 65
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,692
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
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
pysal/mapclassify Provides a set of tools for classifying data to be used in choropleth maps 139