sithu31296/sota-backbones
A collection of SOTA Image Classification Models in PyTorch
This project offers a curated collection of pre-trained image classification models ready for use in various computer vision tasks. You can input your custom image datasets to fine-tune these models for specific classification needs, or integrate them as foundational components into more complex systems like object detection or semantic segmentation. It is designed for machine learning engineers and researchers who need high-performing, state-of-the-art models for image analysis and computer vision development.
169 stars. No commits in the last 6 months.
Use this if you need to quickly implement or experiment with leading image classification models in your computer vision projects, especially for fine-tuning or as backbones for other tasks.
Not ideal if you require distributed training capabilities or specific advanced augmentation techniques like Mixup and Cutmix, which are not currently supported.
Stars
169
Forks
32
Language
Python
License
MIT
Category
Last pushed
May 07, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/sithu31296/sota-backbones"
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