eggie5
/ tensorflow_finetune.py
Created
December 11, 2017 16:15
— forked from omoindrot/tensorflow_finetune.py
Example TensorFlow script for fine-tuning a VGG model (uses tf.contrib.data)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Example TensorFlow script for finetuning a VGG model on your own data. | |
| Uses tf.contrib.data module which is in release v1.2 | |
| Based on PyTorch example from Justin Johnson | |
| (https://gist.github.com/jcjohnson/6e41e8512c17eae5da50aebef3378a4c) | |
| Required packages: tensorflow (v1.2) | |
| Download the weights trained on ImageNet for VGG: | |
| ``` | |
| wget http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz |
eggie5
/ verify_tfrecords.py
Created
August 7, 2019 21:16
— forked from LeegleechN/verify_tfrecords.py
Check TFRecords
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """Checks if a set of TFRecords appear to be valid. | |
| Specifically, this checks whether the provided record sizes are consistent and | |
| that the file does not end in the middle of a record. It does not verify the | |
| CRCs. | |
| """ | |
| import struct | |
| from multiprocessing import Pool | |
| import tensorflow as tf |