search

HunterXuan / wx-tfjs-demo

微信小程序运行 TensorFlow 的 Demo
https://hunterx.xyz/retry-tfjs-on-wx.html
525 stars 100 forks source link

训练好的模型导入到小程序用coco-ssd加载报错 #57

Open shuizhuyu1015 opened 1 year ago

shuizhuyu1015 commented 1 year ago

请问怎么调整训练后转化的模型 与 小程序coco-ssd加载的兼容啊。

我训练了一个模型ssd_mobilenet_v1_fpn_model模型，转化成json后在小程序用coco-ssd加载它，detect报错如下 Tensor must have a shape comprised of positive integers but got shape [100,].

这可能是格式问题，但是不知道怎么调整它，大佬知道吗

HunterXuan commented 1 year ago

应该是输入的 shape 不匹配，可以往这个方向查查

shuizhuyu1015 commented 1 year ago

计算方法上参照您的方法设置的，是不是模型不对，怎样才能导出您那样的模型啊。 image 我的模型json image