TF_FeatureExtraction
TF_FeatureExtraction copied to clipboard
How to use checkpoints of MobileNet, including ckpt.meta, ckpt.index, and ckpt.data
The checkpoints of pre-trained MobileNet are as follows:
mobilenet_v1_1.0_224.ckpt.data-00000-of-00001
mobilenet_v1_1.0_224.ckpt.index
mobilenet_v1_1.0_224.ckpt.meta
rather than one single checkpoint file like:
resnet_v1_101.ckpt
I take a look at the code in feature_extractor.py. It looks that using a directory was not working.
# Find the checkpoint file
checkpoint_path = self._checkpoint_path
if tf.gfile.IsDirectory(self._checkpoint_path):
checkpoint_path = tf.train.latest_checkpoint(self._checkpoint_path)
Could you pls give an advice?
have you solve this problem,i also wang to extract features using checkpoint file, thanks
@jackchowtop1 Not yet. I am still working on. But I can extract features on most models. Pls see https://github.com/zengsn/TF_FeatureExtraction/