0
点赞
收藏
分享

微信扫一扫

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


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

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

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

ls

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

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

cd

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

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

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

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


举报
0 条评论