TF_FeatureExtraction icon indicating copy to clipboard operation
TF_FeatureExtraction copied to clipboard

How to use checkpoints of MobileNet, including ckpt.meta, ckpt.index, and ckpt.data

Open zengsn opened this issue 6 years ago • 2 comments

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?

zengsn avatar Mar 17 '18 14:03 zengsn

have you solve this problem,i also wang to extract features using checkpoint file, thanks

jackchowtop1 avatar Apr 11 '18 03:04 jackchowtop1

@jackchowtop1 Not yet. I am still working on. But I can extract features on most models. Pls see https://github.com/zengsn/TF_FeatureExtraction/

zengsn avatar Apr 12 '18 05:04 zengsn