从Tensorflow Object Detection API动物园模型导出错误的冻结图形

Question

I am working with Mask R CNN model using Tensorflow Object Detection API project ( https://github.com/tensorflow/models/tree/r1.12.0/research/object_detection ). I stick to r1.12.0 release (but this is not a must but I do not think it influences my problem.) My plan is to modify some "static" parts of the model and export it again into the frozen graph format.

As a first step, I meant to regenerate the frozen graph from the checkpoints file and the pipeline.config using export_inference_graph.py script ( https://github.com/tensorflow/models/blob/r1.12.0/research/object_detection/export_inference_graph.py ). I downloaded the inception V2 model ( http://download.tensorflow.org/models/object_detection/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz ) and executed the script using Tensorflow 1.12.0. It does the job and creates a frozen graph.

The issue is that if I compare the original frozen graph with the generated one, they are different. If I visualize them using Tensorboard there are obvious differences between them. Some nodes are missing, some nodes are different etc.

I have tried other models as well (normal Fast R CNN), I had the same issue always.

How can this be? How should I use the checkpoint files and the pipeline.config file to regenerate exactly the same frozen graph which is originally attached?

Answer 1

As far as I understood it, your steps of generating frozen graph is fine.

One thing about frozen graph is that optimizations can be performed on it, for example fuse some layers together. Optimization may cause your frozen graph to look different as different optimizations could be performed or non at all. But different frozen graph does not necessarily mean the graph is wrongly generated.

Here is a tutorial on optimizations on frozen graph to make a faster serving model. Just listed here to show that there are several optimization options.

Here What does freezing a graph in TensorFlow mean? is another problem that is also kind of related to this problem.