Mayurji/Image-Classification-PyTorch
Learning and Building Convolutional Neural Networks using PyTorch
This project helps machine learning practitioners or researchers compare and build image classification models. You input images and train a model, and it outputs a trained model capable of categorizing new images, along with performance metrics like accuracy and loss plots. It's designed for individuals who are experimenting with different deep learning architectures for image recognition tasks.
219 stars. No commits in the last 6 months.
Use this if you are a machine learning practitioner looking to quickly set up, train, and benchmark various image classification models using established architectures like ResNet, VGGNet, or MobileNet.
Not ideal if you need a plug-and-play solution for immediate image classification without diving into model training or hyperparameter tuning.
Stars
219
Forks
52
Language
Python
License
GPL-3.0
Category
Last pushed
Apr 01, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Mayurji/Image-Classification-PyTorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
qubvel/efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras.
KichangKim/DeepDanbooru
AI based multi-label girl image classification system, implemented by using TensorFlow.
matlab-deep-learning/Image-Classification-in-MATLAB-Using-TensorFlow
This example shows how to call a TensorFlow model from MATLAB using co-execution with Python.
AFAgarap/cnn-svm
An Architecture Combining Convolutional Neural Network (CNN) and Linear Support Vector Machine...
harvitronix/five-video-classification-methods
Code that accompanies my blog post outlining five video classification methods in Keras and TensorFlow