如何编辑新的 object-detection-tutorial.ipynb

Question

我想编辑 object_detection_tutorial.ipynb 以使用我训练过的模型，但我在网上遵循的教程有旧的 tensorflow 对象检测 api

我尝试使用旧的 object_detection_tutorial.ipynb 并继续按照教程进行操作，但它没有显示边界框，但是当我尝试他们检测到狗和海滩的原始代码时，它运行良好使用新的 object_detection_tutorial.ipynb 所以决定只编辑新的，但我不能真正正确地做到这一点

这就是我所说的新 object_detection_tutorial.ipynb https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb 我正在关注 2 年前的 senddex 教程，当时可用的 object_detection_tutorial.ipynb 发生了很多变化

我需要在 object_detection_tutorial.ipynb 中进行哪些更改才能使用我自己的训练模型

Answer 1

您应该尝试在下面的 api 中导入您的模型 -

detection_model = load_model(model_name)

这对我有用，希望您已加载地图文件。