linux如何安装caffe

sudo apt-get update
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev
sudo apt-get install -y libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install -y python-dev python-numpy python-pip

git clone https://github.com/BVLC/caffe.git
cd caffe

cp Makefile.config.example Makefile.config

make all -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)

make pycaffe -j$(nproc)
make install -j$(nproc)

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

CUDA_DIR := /usr/local/cuda
USE_CUDNN := 1

make all -j4

pip install -r python/requirements.txt

import caffe
import numpy as np

# 加载模型
model_def = 'path/to/deploy.prototxt'
model_weights = 'path/to/model.caffemodel'
net = caffe.Net(model_def, model_weights, caffe.TEST)

# 准备输入数据
transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})
transformer.set_transpose('data', (2,0,1))
transformer.set_mean('data', np.array([104,117,123])) # 根据模型调整均值
transformer.set_raw_scale('data', 255)
transformer.set_channel_swap('data', (2,1,0))

# 加载并预处理图像
image = caffe.io.load_image('path/to/image.jpg')
transformed_image = transformer.preprocess('data', image)

# 设置输入数据并进行前向传播
net.blobs['data'].data[...] = transformed_image
output = net.forward()

# 输出结果
print(output)