Keras实现的RetinaNet目标检测
Keras RetinaNet Keras 实现 RetinaNet 对象检测,如 Tsung-Yi Lin、Priya Goyal、Ross Girshick、Kaiming He 和 Piotr Dollár 在本文中所述。
培训 可以在此处找到有关如何培训 keras-retinanet 的示例。
用法 对于 Pascal VOC 的训练,请运行:python examples/train_pascal.py
对于 MS COCO 的培训,请运行:python examples/train_coco.py
一般来说,在你自己的数据集上训练的步骤是: 通过调用 keras_retinanet.models.ResNet50RetinaNet 创建一个模型并编译它。
根据经验,已发现以下编译参数运行良好:model.compile( loss={ 'regression' : keras_retinanet.losses.regression_loss, 'classification': keras_retinanet.loss
2021-12-01 16:06:26
2.15MB
机器学习
1