tools/prototxt2bin.py about caffe-mobile HOT 3 CLOSED

#!/usr/bin/env python
import sys
from os import path
import caffe_pb2
from google.protobuf import text_format

if len(sys.argv) < 2:
    print("Usage: ./prototxt2bin.py net.prototxt")

for filename in sys.argv[1:]:
  basename, extname = path.splitext(filename)
  if extname == '.prototxt':
    print("Read Caffe net text from: " + filename)
    params = caffe_pb2.NetParameter()
    with open(filename, 'rt') as f:
      text_format.Merge(f.read(), params)
    print("Write Caffe net binary to: " + basename + '.protobin')
    with open(basename + '.protobin', 'wb') as f:
      f.write(params.SerializeToString())