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的路径改成上面三个文件的新路径


完成!

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



目录
相关文章
|
3月前
|
机器学习/深度学习 API 算法框架/工具
【Tensorflow+keras】Keras API三种搭建神经网络的方式及以mnist举例实现
使用Keras API构建神经网络的三种方法:使用Sequential模型、使用函数式API以及通过继承Model类来自定义模型,并提供了基于MNIST数据集的示例代码。
54 12
|
3月前
|
机器学习/深度学习 API 算法框架/工具
【Tensorflow+keras】Keras API两种训练GAN网络的方式
使用Keras API以两种不同方式训练条件生成对抗网络(CGAN)的示例代码:一种是使用train_on_batch方法,另一种是使用tf.GradientTape进行自定义训练循环。
38 5
|
3月前
|
机器学习/深度学习 存储 PyTorch
【深度学习】Pytorch面试题:什么是 PyTorch?PyTorch 的基本要素是什么?Conv1d、Conv2d 和 Conv3d 有什么区别?
关于PyTorch面试题的总结,包括PyTorch的定义、基本要素、张量概念、抽象级别、张量与矩阵的区别、不同损失函数的作用以及Conv1d、Conv2d和Conv3d的区别和反向传播的解释。
213 2
|
3月前
|
机器学习/深度学习 TensorFlow 算法框架/工具
【Tensorflow+Keras】keras实现条件生成对抗网络DCGAN--以Minis和fashion_mnist数据集为例
如何使用TensorFlow和Keras实现条件生成对抗网络(CGAN)并以MNIST和Fashion MNIST数据集为例进行演示。
45 3
|
6月前
|
机器学习/深度学习 PyTorch 算法框架/工具
基于Pytorch通过实例详细剖析CNN
基于Pytorch通过实例详细剖析CNN
71 1
基于Pytorch通过实例详细剖析CNN
|
API 算法框架/工具
越来越火的tf.keras模型,这三种构建方式记住了,你就是大佬!!!
越来越火的tf.keras模型,这三种构建方式记住了,你就是大佬!!!
122 0
|
Ubuntu TensorFlow 算法框架/工具
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(小数据集)
本例提取了植物幼苗数据集中的部分数据做数据集,数据集共有12种类别,今天我和大家一起实现tensorflow2.X版本图像分类任务,分类的模型使用ResNet50。 通过这篇文章你可以学到: 1、如何加载图片数据,并处理数据。 2、如果将标签转为onehot编码 3、如何使用数据增强。 4、如何使用mixup。 5、如何切分数据集。 6、如何加载预训练模型。
1415 0
ResNet实战:tensorflow2.X版本,ResNet50图像分类任务(小数据集)
|
机器学习/深度学习 算法框架/工具 Python
Keras-3-实例2-多分类问题
Keras-3-实例2-多分类问题
|
机器学习/深度学习 PyTorch 算法框架/工具
ResNet结构的pytorch实现
ResNet结构的pytorch实现
175 0
ResNet结构的pytorch实现
|
机器学习/深度学习 API 算法框架/工具
Keras创建ANN模型的四种方法
这里以MNIST数据集来介绍Keras创建人工神经网络模型的四种方法
351 0
Keras创建ANN模型的四种方法
下一篇
无影云桌面