automl
automl copied to clipboard
Cannot export a frozen graph which has only a neural network nodes
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:
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
Bumping this- I am having the same problem. @Smokemachine Did you ever figure out the bug?