测试一个训练好的caffe模型

在学习caffe的过程中,训练了出了模型出来,出了当时的准确率和loss值,并没有看到给定输入看到真正的输出,这个时候需要测试一下训练出来的模型,实际查看一下效果,其中用到的配置文件和网络模型在caffe的目录下都有,自己测试自己的模型时需要修改为自己的*.prototxt和*.caffemodel

#!/usr/bin/env python
#coding=utf-8
#因为需要sudo权限,只能作如下处理,自己添加caffe的位置,编译之后,然后用sudo权限运行该.py程序
import sys
caffe_root='/home/dyh/caffe/caffe/'
sys.path.insert(0, caffe_root+'python')
import caffe
import numpy as np
'''
这个测试模型可以帮助自己测试自己训练出来的模型效果如何
'''
caffe.set_mode_gpu()
# caffe.set_device(0)

#deploy文件就是用来测试训练好的网络的,给其输入,自己写测试来输出类别
model_def = '/home/dyh/caffe/caffe/models/bvlc_reference_caffenet/deploy.prototxt'
# model_def = '/home/dyh/caffe-workspace/face_detect/deploy_full_conv.prototxt'
model_weights = '/home/dyh/caffe-workspace/caffe_case/caffe_case模板/bvlc_reference_caffenet.caffemodel'
# model_weights = '/home/dyh/c
  • 0
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值