Ubuntu16.04使用tensorflow压缩facenet网络训练的模型

  1. 压缩使用facenet训练好并冻结以后的模型  
  2. facenet网络:https://github.com/davidsandberg/facenet  
  3. 安装bazel和必要的编译步骤:ubuntu16.04tensorflow压缩模型transform_graph方式  
  4. 在训练好facenet网络,并使用freeze_graph.py冻结模型以后,对于输出的文件facenet_freeze.pb,经过多次尝试,终于压缩好。  
  5. 压缩的方式:  
  6. 各种选项解释:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms  
  7. bazel-bin/tensorflow/tools/graph_transforms/transform_graph \  
  8. --in_graph=/home/用户名/facenet_freeze.pb \  
  9. --out_graph=/home/用户名/optimized_facenet.pb  \  
  10. --inputs="input" \  
  11. --outputs="embeddings"  \  
  12. --transforms='add_default_attributes \  
  13.              fold_constants(ignore_errors=true)  \  
  14.              fold_batch_norms \  
  15.              fold_old_batch_norms \  
  16.              quantize_weights   
  17.              sort_by_execution_order'  

你可能感兴趣的:(tensorflow,深度学习,人工智能)