运用训练好的模型，实现分类（caffe）

最新推荐文章于 2024-06-24 11:12:22 发布

superxiaoying

最新推荐文章于 2024-06-24 11:12:22 发布

阅读量913

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/nuannuanyingying/article/details/74779720

版权

本文介绍如何在已训练好的Caffe模型基础上，进行分类任务。内容包括处理mean.binaryproto文件转换为mean.npy，以及在特定环境下部署模型的注意事项。

摘要由CSDN通过智能技术生成

前言：
0. 模型已经训练
1. 如果训练时进行了均值化处理（就是说你是用mean.binaryproto，这样可以加快训练速度并且提高实现的转确度），那么在生成的时候也需要得到mean.npy

code1:mean.binaryproto –> mean.npy

#!/usr/bin/env python
import numpy as np
import sys,os
caffe_root = 'your caffe root'
sys.path.insert(0, caffe_root + 'python')
import caffe
os.chdir(caffe_root)

if len(sys.argv)!=3:
    print "Usage: python convert_mean.py mean.binaryproto mean.npy"
    sys.exit()

blob = caffe.proto.caffe_pb2.BlobPro