joelthchao / Tensorflow Finetune Flickr Style
Demonstrate the finetune process on TensorFlow
Stars: ✭ 132
Programming Languages
python
139335 projects - #7 most used programming language
Labels
tensorflow-finetune-flickr-style
In this project, we use flickr style dataset to demonstrate finetune in TensorFlow. The details please refer to the example from the Caffe website
Thank @ethereon and @sergeyk for their code. We modify the network.py from caffe-tensorflow and flickr.py from vislab for our use.
Download flickr style dataset
# Download dataset
$ python assemble_data.py images train.txt test.txt 500
Download the pre-trained model
Download link: here
Or follow the tutorial and extract bvlc_alexnet.npy from https://github.com/ethereon/caffe-tensorflow
Lauch the finetune process
$ python finetune.py train.txt test.txt bvlc_alexnet.npy
Finetune result
// Fine-tuning result
Iter 1280: Testing Accuracy = 0.3250
// From scratch result
Iter 1280: Testing Accuracy = 0.1655
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].