Skip to content
Snippets Groups Projects
Select Git revision
  • 12b4c9e6d8deb441acb5da7dfd502a56f6fa92aa
  • master default protected
  • updated_emam2middleware
  • patch-1
  • corrected_get_torcs_image
  • develop
  • multi_variant_model
7 results

binaryproto_to_numpy.py

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    binaryproto_to_numpy.py 483 B
    import caffe
    import csv
    import numpy as np
    from PIL import Image
    import sys
    
    blob = caffe.proto.caffe_pb2.BlobProto()
    data = open("./driving_mean_1F.binaryproto" , 'rb' ).read()
    blob.ParseFromString(data)
    data = np.array(blob.data)
    arr = np.array( caffe.io.blobproto_to_array(blob) )
    arr = arr[0]
    # np.save("./mean_image.npy", arr[0]) # shape is (210, 280, 3)
    
    np.savetxt("./mean_image_R.txt", arr[0])
    np.savetxt("./mean_image_G.txt", arr[1])
    np.savetxt("./mean_image_B.txt", arr[2])