使用自己的图像(张量流)重新训练对象检测模型



早上好,

我一直在使用 tensorflow 对象检测教程,使用它们作为冻结图提供的ssd_mobilenet以及相应的检查点文件(model.ckpt.data-00000-of-00001、model.ckpt.index、model.ckpt.meta(。 但是,由于图像有时识别度很差,我希望我可以将自己的图像提供给检测模型,并提高其图像的性能,这些图像都是由同一台相机拍摄的。

谷歌无法帮助我从哪里开始。我遇到的问题: - 是否有任何代码片段显示要加载哪些文件以及如何训练现有模型? - 我是否需要使用旧数据(即 COCO(+ 新数据(我的图像(重新训练加载的模型,或者我是否可以使用我的数据重新训练它并且模型会记住它以前学到的东西?

很抱歉这个非常不具体的问题,但我只是不知道从哪里开始。

Dat Tran 编写了一个很棒的演练博客和代码库。他训练了一个模型,以预先训练的SSD_mobilenet为起点来识别图像中的浣熊。这是我找到的最好的起点。希望这有帮助。

最新更新