automl icon indicating copy to clipboard operation
automl copied to clipboard

Cannot export a frozen graph which has only a neural network nodes

Open Smokemachine opened this issue 4 years ago • 1 comments

Hi,

I am trying to export a frozen graph which has only a neural network nodes. To do this, I turn on the export mode and set the only_network flag

python inspector.py --mode=export --only_network=true --model_name=efficientdet-d0 --saved_model_dir=saved_model_dir

As a result of the script, a directory with saved_model and protobuf file is created.

saved_model_dir/efficientdet-d0_frozen.pb

After that, I launch the graph for execution and get the following result. Identity, Identity_1, Identity_2, Identity_3, Identity_4 nodes always have the same result, regardless of input data:

image

It seems there is a bug in the export algorithm.

Input files are loaded by executing the following commands

export MODEL=efficientdet-d0 export CKPT_PATH=efficientdet-d0 wget https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/${MODEL}.tar.gz wget https://user-images.githubusercontent.com/11736571/77320690-099af300-6d37-11ea-9d86-24f14dc2d540.png -O img.png tar xf ${MODEL}.tar.gz

Smokemachine avatar Feb 15 '21 10:02 Smokemachine

Bumping this- I am having the same problem. @Smokemachine Did you ever figure out the bug?

cnmckee avatar Mar 26 '21 21:03 cnmckee