Ugenteraan/ResNet-50-CBAM-PyTorch
Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https://www.kaggle.com/puneet6060/intel-image-classification.
This helps deep learning engineers train and evaluate image classification models more effectively. You provide a dataset of categorized images, and it outputs a trained model ready to classify new images, potentially with improved accuracy by using the CBAM attention mechanism. This tool is for machine learning practitioners focused on computer vision tasks.
100 stars. No commits in the last 6 months.
Use this if you are a deep learning engineer needing a pre-built ResNet-50 architecture to classify images, especially if you want to experiment with or leverage a CBAM attention module for better performance.
Not ideal if you are a business user looking for a no-code image classification solution or if you need to classify non-image data.
Stars
100
Forks
23
Language
Python
License
MIT
Category
Last pushed
Jan 11, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Ugenteraan/ResNet-50-CBAM-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