#635 · Primary category: Computer Vision

finetune_alexnet_with_tensorflow

alexnet deep-learning fine-tune tensorflow

Code for finetuning AlexNet in TensorFlow >= 1.2rc0

Project last updated:03/05/19

GitHub Stars

777

Forks

401

Contributors

2

License

BSD-3-Clause

Why we included this project

Finetuning AlexNet on your own image dataset is the whole point of this repo, and the code is small enough to read end to end. You get the full AlexNet graph definition, a finetune script that takes plain text files of image paths and class labels, and a data generator wrapper for TensorFlow's input pipeline, plus a notebook for checking the pretrained weights against ImageNet images. That makes it a solid teaching example for how transfer learning is wired up, even if the TensorFlow API and the architecture itself are older than what teams would pick today.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category