Download pretrained ImageNet model of ResNet, VGG, etc. (.PB file)

python tensorflow pre-trained-model
Ajay Maity picture Ajay Maity · May 6, 2018 · Viewed 24.5k times · Source

I have downloaded a pre-trained model on ImageNet of Inception v3 from http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz (found this link while following one of the tutorials on codelabs).

This zip file contains .pb file that I can directly import as GraphDef in TensorFlow.

I would like to know if there are similar .pb files for other architectures as well, such as ResNet, VGG16, MobileNet, etc. If yes, could you provide the link of those?

Thanks in advance.

Kind Regards,

Ajay

Answer

walkerlala picture walkerlala · May 7, 2018

You can find many pretrained models here: https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

And the corresponding codes are here: https://github.com/tensorflow/models/tree/master/research/slim/nets

Related questions

AttributeError: 'Node' object has no attribute 'output_masks'

I use Keras pretrained model VGG16. The problem is that after configuring tensorflow to use the GPU I get an error that I didn't have before when using the CPU. The error is the following one: Traceback (most recent call …

Read More
How to save/load a tensorflow hub module to/from a custom path?

The tensorflow_hub library maintainers has made it every easy for users to download and use the pre-trained tensorflow modules, e.g.: import tensorflow_hub as hub embed = hub.Module("https://tfhub.dev/google/universal-sentence-encoder/1") But from the sys.stderr …

Read More
How to find which version of TensorFlow is installed in my system?

I need to find which version of TensorFlow I have installed. I'm using Ubuntu 16.04 Long Term Support.

Read More