用caffe训练好的lenet_iter_10000.caffemodel测试单张mnist图片

最新推荐文章于 2022-11-06 20:29:32 发布

_苏_

最新推荐文章于 2022-11-06 20:29:32 发布

阅读量1.5w

点赞数 4

分类专栏：深度学习文章标签： caffemodel classify mnist

本文链接：https://blog.csdn.net/lanxuecc/article/details/52485077

版权

本文介绍了如何利用已训练好的lenet_iter_10000.caffemodel对MNIST数据集的一张手写数字图片进行测试。首先，准备一张手写数字图片并确保其通道数正确。接着，由于训练模型未计算均值文件，创建一个全0的均值文件。然后，修改classify.py为classifymnist.py，并执行测试命令，最终得到了手写数字的识别结果。

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

接上个博客：http://blog.csdn.net/lanxuecc/article/details/52474476，前面已经生成了deploy.proto.txt。下面具体记录用lenet_iter_10000.caffemodel测试图片。

准备一张手写数字图片

这里写图片描述

注意在depoy.prototxt文件中指定正确的该图片的通道数。

准备一个均值文件

因为classify.py中的测试接口caffe.Classifier需要训练图片的均值文件作为输入参数，而实际lenet-5训练时并未计算均值文件，所以这里创建一个全0的均值文件输入。编写一个zeronp.py文件如下
这里写图片描述
执行

python zeronp.py

生成均值文件 meanfile.npy。
这里注意宽高要与输入测试的图片宽高一致。这里参考：https://github.com/BVLC/caffe/issues/320

修改classify.py保存为classifymnist.py文件

#!/usr/bin/env python
"""
classify.py is an out-of-the-box image classifer callable from the command line.

By default it configures and runs the Caffe reference ImageNet model.
"""
import numpy as np
import os
import sys
import argparse
import glob
import time
import pandas as pd #插入数据分析包

import caffe

def main(argv):
    pycaffe_dir = os.path.dirname(__file__)

    parser = argparse.ArgumentParser()
    # Required arguments: input and output files.
    parser.add_argument(
        "input_file",
        help="Input image, directory, or npy."
    )
    parser.add_argument(
        "output_file",
        help="Output npy filename."
    )
    # Optional arguments.
    parser.add_argument(
        "--model_def",
        default=os.path.join(pycaffe_dir,
                "../examples/mnist/deploy.prototxt"), #指定lenet-5的deploy.prototxt模型位置
        help=