python - 如何使用自定义数据集运行 Keras Retinanet 对象检测代码示例？

Question

我正在尝试从源代码实现 Keras Retinanet 示例：

https://keras.io/examples/vision/retinanet/

而且我确实让示例与 COCO 数据集完美运行。现在我想使用我自己的自定义对象检测数据集运行该示例。在网上做了几天的研究后，我仍然不清楚，我需要如何编辑示例代码以使用我自己的数据集（即一组 .jpg 图像 + 注释 .xml 文件标签图像）。

我怀疑我需要熟悉tensorflow_dataset-library吗？我的猜测是代码中的这一部分是我应该为我自己的数据集进行自定义的地方：

https://keras.io/examples/vision/retinanet/#load-the-coco2017-dataset-using-tensorflow-datasets

为了让 Keras Retinanet 对象检测示例与我自己的数据集一起工作，有什么建议或好的参考资料值得研究吗？

Answer 1

我找到了一个用于实施 Retinanet 的精心指导的协作链接： https ://colab.research.google.com/github/benihime91/pytorch_retinanet/blob/master/demo.ipynb

还有另一个链接： https ://www.programmersought.com/article/73514120308/

上面两个粘贴的链接都得到了很好的描述，并且适用于自定义数据集。