How to create frozen graph proto file from trained model

ruiminshen / yolo-tf

TensorFlow implementation of the YOLO (You Only Look Once)

GNU Lesser General Public License v3.0

198 stars 72 forks source link

How to create frozen graph proto file from trained model #12

Open alexanderfrey opened 6 years ago

alexanderfrey commented 6 years ago

Hello,

thanks for your excellent work. I'm currently trying to learn a pretrained yolo2 model with my own images and export the model as a frozen graph.

Can you shed some light on how to do that and where to find the the pipeline config that is needed for the export_inference_graph.py from obejct detection api ?

Thanks in advance !

Alexander