gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 13:25:53 29 4
gpt4 key购买 nike

我正在尝试从源代码实现 Keras Retinanet 示例:
https://keras.io/examples/vision/retinanet/
我确实让示例与 COCO 数据集完美运行。现在我想用我自己的自定义对象检测数据集运行这个例子。在网上做了几天的研究后,我仍然不太清楚,我需要如何编辑示例代码以使用我自己的数据集(这是一组 .jpg 图像 + 注释 .xml 文件,用labelImg)。
我怀疑我需要熟悉 tensorflow_dataset-library ?我的猜测是代码中的这部分是我应该为自己的数据集进行自定义的地方:
https://keras.io/examples/vision/retinanet/#load-the-coco2017-dataset-using-tensorflow-datasets
为了让 Keras Retinanet 对象检测示例与我自己的数据集一起工作,是否有任何值得研究的建议或好的引用资料?

最佳答案

我找到了一个很好的指导协作链接,用于实现 Retinanet:
https://colab.research.google.com/github/benihime91/pytorch_retinanet/blob/master/demo.ipynb
另外还有一个链接:
https://www.programmersought.com/article/73514120308/
上面粘贴的两个链接都有很好的描述,适用于自定义数据集。

关于python - 如何使用自定义数据集运行 Keras Retinanet 对象检测代码示例?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68573265/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com