creafz/pytorch-cnn-finetune
Fine-tune pretrained Convolutional Neural Networks with PyTorch
This tool helps machine learning engineers and researchers quickly adapt powerful, pre-trained image recognition models for new image classification tasks. You input your image dataset and it outputs a specialized model ready for training, even if your images are a different size or have a unique set of categories. It's designed for those who work with deep learning and want to leverage existing neural networks efficiently.
725 stars. No commits in the last 6 months.
Use this if you need to classify images and want to significantly speed up model development by starting with state-of-the-art convolutional neural networks instead of building one from scratch.
Not ideal if your task does not involve image data or if you need to develop a neural network architecture completely from the ground up without using pre-trained components.
Stars
725
Forks
121
Language
Python
License
MIT
Category
Last pushed
Aug 16, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/creafz/pytorch-cnn-finetune"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
kratzert/finetune_alexnet_with_tensorflow
Code for finetuning AlexNet in TensorFlow >= 1.2rc0
bamos/densenet.pytorch
A PyTorch implementation of DenseNet.
raghakot/keras-resnet
Residual networks implementation using Keras-1.0 functional API
liuzhuang13/DenseNet
Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).
gpleiss/efficient_densenet_pytorch
A memory-efficient implementation of DenseNets