vgg16 on keras for tensorflow
2017-06-21 17:33
507 查看
测试用
jpeg
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
相关文章推荐
- A Newbie’s Install of Keras & Tensorflow on Windows 10 with R
- MAP@K metric for keras with tensorflow backend
- TensorFlow for Hackers (Part VI) - Human Activity Recognition using LSTMs on Android
- TensorFlow for Hackers (Part VII) - Credit Card Fraud Detection using Autoencoders in Keras
- 翻译 | Keras : Deep Learning library for Tensorflow and Theano
- Deploying Tensorflow model on Andorid device for Human Activity Recognition
- GAN by Example using Keras on Tensorflow Backend
- tensorflow75 Keras on Tensorflow 入门
- Retrain a tensorflow model based on Inception v3
- 如何落地TensorFlow on Kubernetes
- Keras/Tensorflow 在 Mac 简易安装
- Training for Tensorflow object detection API
- Notes on tensorflow(六)variable_scope
- Notes on tensorflow(七)将数据集转换为TFRecord
- Keras设定GPU使用内存大小(Tensorflow backend)
- Installing TensorFlow on Ubuntu -转载
- 雅虎开源TensorflowOnSpark,Ubuntu配置安装TensorflowOnSpark
- tensorflow TypeError: run() got multiple values for argument 'feed_dict'
- Tensorflow for Machine Intelligence 读后
- 深度工具合集安装(Nvidia+CUDA+cuDNN+Tensorflow+OpenBLAS+Caffe+Theano+Keras+Torch+Mxnet+X2Go)