Caffe使用

[zixie1991]

Caffe使用

记录Caffe使用过程中遇到的问题及解决办法

[zixie1991]

pycaffe

memory_data_layer

datum = caffe.proto.caffe_pb2.Datum()
datum.ParseFromString(dat)
arr = np.array(caffe.io.datum_to_array(datum))
env.close()

# assign input
net.add_datum_vector(np.array([arr], dtype=np.float32),
                     np.array([0], dtype=np.float32))

# run
net._forward(0, len(net.layers) - 1)

# print result
print net.blobs[net.outputs[-1]].data.squeeze(axis=(2,3))