keras vgg16模型第一次使用注意事项

简介: 用官网例子: from import from import from import import as weights'imagenet'include_topTrue'./timg.jpg'target_size224224axis0 print('Predicted:',decode_p.

用官网例子:


 

from keras.applications.vgg16 import VGG16


from keras.preprocessing import image


from keras.applications.vgg16 import preprocess_input,decode_predictions


import numpy as np

model = VGG16(
weights = 'imagenet' , include_top = True )

img_path =
'./timg.jpg'

img = image.load_img(img_path, target_size =( 224 , 224 ))
x = image.img_to_array(img)
x = np.expand_dims(x,
axis = 0 )
x = preprocess_input(x)

preds = model.predict(x)

print('Predicted:',decode_predictions(preds,top=3)[0])




会报错

解决办法:

1、按报错提供的网址下载vgg16_weights_tf_dim_ordering_tf_kernels.h5(include_top=True

vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5(include_top=False

以及imagenet_class_index.json(imagenet_utils. decode_predictions


2、输入 open .keras/models/ 打开影藏文件夹models,将上面三个文件放进去


3、print(keras.__file__),找到安装路径,在applications里面找到vgg.py和imagenet_utils.py,

WEIGHTS_PATH,WEIGHTS_PATH_NO_TOP,CLASS_INDEX_PATH的路径改成上面三个文件的新路径


完成!

其他模型方法应该是一样的



目录
相关文章
|
2月前
|
机器学习/深度学习 PyTorch 算法框架/工具
基于Pytorch通过实例详细剖析CNN
基于Pytorch通过实例详细剖析CNN
33 1
基于Pytorch通过实例详细剖析CNN
|
9月前
|
API 算法框架/工具
越来越火的tf.keras模型,这三种构建方式记住了,你就是大佬!!!
越来越火的tf.keras模型,这三种构建方式记住了,你就是大佬!!!
|
机器学习/深度学习 PyTorch 算法框架/工具
【Pytorch神经网络实战案例】02 CIFAR-10数据集:Pytorch使用GPU训练CNN模版-方法②
【Pytorch神经网络实战案例】02 CIFAR-10数据集:Pytorch使用GPU训练CNN模版-方法②
231 0
|
Ubuntu TensorFlow 算法框架/工具
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(小数据集)
本例提取了植物幼苗数据集中的部分数据做数据集,数据集共有12种类别,今天我和大家一起实现tensorflow2.X版本图像分类任务,分类的模型使用ResNet50。 通过这篇文章你可以学到: 1、如何加载图片数据,并处理数据。 2、如果将标签转为onehot编码 3、如何使用数据增强。 4、如何使用mixup。 5、如何切分数据集。 6、如何加载预训练模型。
1329 0
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(小数据集)
|
TensorFlow 算法框架/工具
TensorFlow自定义损失函数
TensorFlow自定义损失函数
94 0
|
机器学习/深度学习 算法 数据可视化
基于PaddlePaddle框架对CIFAR-100数据集在简易CNN(LeNet-5修改)和简易DNN的效果对比
基于PaddlePaddle框架对CIFAR-100数据集在简易CNN(LeNet-5修改)和简易DNN的效果对比
260 0
基于PaddlePaddle框架对CIFAR-100数据集在简易CNN(LeNet-5修改)和简易DNN的效果对比
|
TensorFlow 算法框架/工具 计算机视觉
ResNet实战:tensorflow2.0以上版本,使用ResNet50实现图像分类任务
ResNet实战:tensorflow2.0以上版本,使用ResNet50实现图像分类任务
739 0
|
机器学习/深度学习 PyTorch 算法框架/工具
ResNet结构的pytorch实现
ResNet结构的pytorch实现
135 0
ResNet结构的pytorch实现
|
机器学习/深度学习 API 算法框架/工具
Keras创建ANN模型的四种方法
这里以MNIST数据集来介绍Keras创建人工神经网络模型的四种方法
269 0
Keras创建ANN模型的四种方法
|
人工智能 算法 Ubuntu
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(大数据集)
本例提取了猫狗大战数据集中的部分数据做数据集,演示tensorflow2.X版本如何使用Keras实现图像分类,分类的模型使用ResNet50。本文实现的算法有一下几个特点: 1、自定义了图片加载方式,更加灵活高效,不用将图片一次性加载到内存中,节省内存,适合大规模数据集。 2、加载模型的预训练权重,训练时间更短。 3、数据增强选用albumentations。
1389 0
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(大数据集)