Caffe Model Zoo

最新推荐文章于 2025-10-27 10:59:58 发布

原创最新推荐文章于 2025-10-27 10:59:58 发布 · 2.8k 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍如何使用Caffe工具包下载并部署预训练的caffenet模型进行图像分类任务。通过一系列命令行操作，包括下载模型文件、加载模型，并对一张猫的图片进行预测，展示模型的分类效果。

Caffe Model Zoo

下载meta数据到当前目录：

./get_ilsvrc_aux.sh

下载caffenet模型：

cd ../../models/nclv_reference_caffenet/

wget http://dl.caffe.berkeleyvision.org/bvlc_reference_caffenet.caffemodel

回到根目录
./build/examples/cpp_classification/classification.bin models/bvlc_reference_caffenet/deploy.prototxt models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel data/ilsvrc12/imagenet_mean.binaryproto data/ilsvrc12/synset_words.txt examples/images/cat.jpg

输出如下所示：
———- Prediction for examples/images/cat.jpg ———-
0.3134 - “n02123045 tabby, tabby cat”
0.2380 - “n02123159 tiger cat”
0.1235 - “n02124075 Egyptian cat”
0.1003 - “n02119022 red fox, Vulpes vulpes”
0.0715 - “n02127052 lynx, catamount”