将具有权重不匹配的keras模型转换为tensorflowjs,groups = 2的层

如何解决将具有权重不匹配的keras模型转换为tensorflowjs,groups = 2的层

我正在构建一个keras模型,将保留的权重加载到该模型上,然后尝试将其转换为tfjs,问题出在将groups参数设置为2的一层。我将权重形状与预先转换的形状和keras模型可以匹配,但是当我将其转换为tfjs并加载到brower上时,会出现不匹配的错误。

代码:

eyes_model = keras.Sequential(
[
keras.Input(shape=(224,224,3)),layers.Conv2D(96,kernel_size=11,strides=4,padding='valid',activation='relu'),layers.MaxPooling2D(pool_size=3,strides=2),layers.Conv2D(256,kernel_size=5,strides=1,padding='same',groups=2,# This is the layer mismatch happens
layers.MaxPooling2D(pool_size=3,layers.Conv2D(384,kernel_size=3,layers.Conv2D(64,kernel_size=1,activation='relu')
]
)

正确的权重应该是:[5、5、48、256],这是预先训练的权重的形状,但是当我将其转换并加载到浏览器中时,错误就会发生:

Uncaught (in promise) Error: Shape mismatch: [5,5,96,256] vs. [5,48,256]
at checkShapesMatch (webgazer.js:66610)
at variables_LayerVariable.write (webgazer.js:66578)
at webgazer.js:66781
at Array.forEach ()
at batchSetValue (webgazer.js:66779)
at models_Sequential.loadWeights (webgazer.js:70579)
at loadLayersModelFromIOHandler (webgazer.js:73987)

通过以下方式完成转换: tfjs.converters.save_keras_model(model)

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com(将#修改为@)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?
Java在半透明框架/面板/组件上重新绘画。
Java“ Class.forName()”和“ Class.forName()。newInstance()”之间有什么区别?
在此环境中不提供编译器。也许是在JRE而不是JDK上运行?
Java用相同的方法在一个类中实现两个接口。哪种接口方法被覆盖?
Java 什么是Runtime.getRuntime()。totalMemory()和freeMemory()?
java.library.path中的java.lang.UnsatisfiedLinkError否*****。dll
JavaFX“位置是必需的。” 即使在同一包装中
Java 导入两个具有相同名称的类。怎么处理?
Java 是否应该在HttpServletResponse.getOutputStream()/。getWriter()上调用.close()?
Java RegEx元字符(。)和普通点?