java 深度学习 图像分类_OpenCV3与深度学习实例-使用Caffe训练好的GoogLeNet模型进行图片分类识别...

#coding:utf-8

import cv2 as cv

import time

import numpy as np

def predict(image_path):

prototxt = 'datas/models/caffe/bvlc_googlenet.prototxt'

caffemodel = 'datas/models/caffe/bvlc_googlenet.caffemodel'

synsetwords = 'datas/models/caffe/bvlc_googlenet_synset_words.txt'

image = cv.imread(image_path)

image = cv.resize(image,(256,256))

# 加载分类

rows = open(synsetwords).read().strip().split("\n")

classes = [r[r.find(" ") + 1:].split(",")[0] for r in rows]

# 根据 Deploy 的 Prototxt, 得知训练后的 CNN model 需要输入数据的维度为

# (1, 3, 224, 224)

# 透过训练时的 Prototxt, 可以得知训练集的 Pixel RGB 平均值为 (104, 117, 123)

# https://github.com/BVLC/caffe/blob/master/models/bvlc_googlenet/train_val.prototxt

# 做 Mean subtraction 的目的 : 改善对抗光线改变造成的影响

blob = cv