trainCaffe.py 文件源码

def printNetwork(prototxt_filename, caffemodel_filename):
    '''
    Draw the CNN architecture as you want, actually, it may not work property
    '''
    _net = caffe.proto.caffe_pb2.NetParameter()
    f = open(prototxt_filename)
    google.protobuf.text_format.Merge(f.read(), _net)
    caffe.draw.draw_net_to_file(_net, prototxt_filename + '.png' )
    print('Draw CNN finished!')