您的位置:首页 > 其它

vgg16 on keras for tensorflow

2017-06-21 17:33 507 查看
测试用

from keras.models import Sequential
from keras.layers.core import Flatten, Dense, Dropout
from keras.layers.convolutional import Convolution2D, MaxPooling2D, ZeroPadding2D
from tensorflow.contrib.keras.python.keras.optimizers import SGD
import  numpy as np


Using TensorFlow backend.


import cv2
%pwd


'D:\\ai'


from tensorflow.contrib.keras.python.keras.applications.vgg16 import VGG16


model = VGG16('vgg16_weights.h5')
sgd = SGD(lr=0.1, decay=1e-6, momentum=0.9, nesterov=True)
#model.compile(optimizer=sgd, loss='mean_squared_error')
model.compile(loss='mean_squared_error', optimizer=sgd)


def load_image(imageurl):
im = cv2.resize(cv2.imread(imageurl),(224,224)).astype(np.float32)
#im[:,:,0] -= 103.939
#im[:,:,1] -= 116.779
#im[:,:,2] -= 123.68
#im = im.transpose((2,0,1))
im = np.expand_dims(im,axis=0)
return im


f = open('synset_words.txt','r')
lines = f.readlines()
f.close()


def predict(url):
im = load_image(url)
pre = np.argmax(model.predict(im))
print(lines[pre])


%pylab inline


Populating the interactive namespace from numpy and matplotlib

D:\Anaconda3\lib\site-packages\IPython\core\magics\pylab.py:161: UserWarning: pylab import has clobbered these variables: ['f']
`%matplotlib` prevents importing * from pylab and numpy
"\n`%matplotlib` prevents importing * from pylab and numpy"


from IPython.display import Image


Image('airplane.jpg')


jpeg

predict('liuyan.jpg')


n10148035 groom, bridegroom





                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: