欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu上将Imagenet的VGG、ResNet、Densenet、Inception等模型参数导入tensorfow里面

程序员文章站 2024-03-15 09:21:59
...

首先,你有了参数文件(像这样的,如图),没有的话可以参考我的另一篇博客下载

ubuntu上将Imagenet的VGG、ResNet、Densenet、Inception等模型参数导入tensorfow里面

然后打开终端,输入ls -a,查看隐藏文件

ls -a

ubuntu上将Imagenet的VGG、ResNet、Densenet、Inception等模型参数导入tensorfow里面

可以看到有个.keras隐藏文件,进去cd .keras

cd .keras

里面有个models文件,将.h5文件放到这里面就算导入成功

ubuntu上将Imagenet的VGG、ResNet、Densenet、Inception等模型参数导入tensorfow里面

最后只需要用cp或mv命令把文件移过来即可,再去执行以下代码就没问题了

model = keras.applications.VGG19(weights='imagenet', include_top=False,
                               pooling='max')