我正在尝试从源代码实现 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 对象检测示例与我自己的数据集一起使用？
我正在使用 anaconda python 3.7 和tensorflow 2.1以及cuda 10.1和cudnn 7.6.5，并尝试运行retinaset（https://github.com/fizyr/keras-retinanet）：
python keras_retinanet/bin/train.py --freeze-backbone --random-transform --batch-size 8 --steps 500 --epochs 10 csv annotations.csv classes.csv
下面是由此产生的错误：
Epoch 1/10
2020-02-10 20:34:37.807590: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-02-10 20:34:38.835777: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2020-02-10 20:34:39.753051: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2020-02-10 20:34:39.776706: W tensorflow/core/common_runtime/base_collective_executor.cc:217] BaseCollectiveExecutor::StartAbort Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if …