使用 MobileNet 重新训练图像检测 [英] Retrain image detection with MobileNet

问题描述

重新训练 MobileNet 以与 Tensorflow.js 一起使用的几种方法对我来说都失败了.有没有办法在 Tensorflow.js 中使用重新训练的模型?

Several ways of retraining MobileNet for use with Tensorflow.js have failed for me. Is there any way to use a retrained model with Tensorflow.js?

使用现代的、基于中心的教程以及使用 retrain.py 似乎都失败了.

Both using the modern, hub-based tutorial, as well as using retrain.py seem to fail.

• 将 retrain.py 的输出转换为 tensorflow.js
• 将 keras 模型转换为 tfjs 时出错:重复权重名称变量

以及其他一些未解决的问题

as well as some other open questions

• 在 tensorflow js 中重新训练图像分类器
• 加载 mobilenet v2 有效，但预训练的 mobilenetv2 失败
• 无法将 TensorFlow 保存的模型转换为 tfjs_layers_model网络模型

前两个问题显示了在这两种情况下都失败的代码，都没有解决.

The top two other questions show the code that failed in both instances, both are unsolved.

目的是加载 mobilenet，使用自定义数据重新训练，并在 Tensorflow.js 中使用它.遵循这两个教程似乎都失败了.这可以在 node.js 中完成吗?还有其他方法吗?我在哪里犯了错误(或者软件无法使用重新训练的模型)?这如何工作?

The aim is to load the mobilenet, retrain using custom data, and use it in Tensorflow.js. Following both tutorials seem to fail. Could this be done inside node.js? Is there another way? Where did I make mistakes (or is the software unable to use retrained models)? How can this work?

最新的 github 问题和另外一个问题

推荐答案

我遇到了同样的问题，似乎我们使用了错误的方法.TF转换模型有loadGraphModel，Keras有loadLayersModel我对这个问题的评论

I encountered the same problem and it seems that we use the wrong method. There are loadGraphModel for TF converted models and loadLayersModel for Keras ones my comment about the issue

这篇关于使用 MobileNet 重新训练图像检测的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！