我遇到了这个存储库：
https://github.com/ArunMichaelDsouza/tensorflow-image-detection利用了tensorflow的示例代码：
https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py
我曾尝试修改retrain.py以使其导出SavedModel (saved_model.pb and a folder named variables)以转换为客户端javascript模型。 (tfjs_graph_model)
经过数小时的研究，我首先发现我无法转换冻结的模型，而导出SavedModel的唯一方法是使用内置的SavedModelBuilder的tensorflow，但我不知道如何合并到代码中。
我看过像
这样的例子
https://github.com/tensorflow/serving/blob/master/tensorflow_serving/example/mnist_saved_model.py
但我什至不知道x文件中y和retrain.py变量的位置。
我的目标是使它导出SavedModel，以便我可以使用tensorflowjs_converter将其转换为tfjs_graph_model
编辑：我对python和tensorflow（不是javascript tho）非常了解，因此请回答retrain.py文件的代码片段，以使用SavedModel导出SavedModelBuilder以及每个步骤的注释。谢谢！ ^。^