Commit 287fb5dc authored by Max Mizikar's avatar Max Mizikar
Browse files

Correct path to mean

parent 59648e48
Pipeline #168 skipped
......@@ -32,7 +32,7 @@ PRETRAINED = './shuffled_iter_119000.caffemodel'
IMAGE_FOLDERS = './classify/*'
caffe.set_mode_gpu()
mean_filename = '/home/ubuntu/gender/mean_image/Test_folder_is_4/mean.binaryproto'
mean_filename = './mean.binaryproto'
proto_data = open(mean_filename, "rb").read()
a = caffe.io.caffe_pb2.BlobProto.FromString(proto_data)
mean = caffe.io.blobproto_to_array(a)[0]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment