如何使用自定义数据集运行 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/

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