如何解决使用 Tensorflow 对象检测 API 从 Imagenet 分类特征提取器开始训练检测模型
TF 对象检测 API 在 COCO 检测数据集上提供预训练模型。但是,我找不到任何关于特征提取器(EfficientNet、MobileNet、ResNet50 等)的预训练检查点的参考。
我假设这些 COCO 模型是从在 ImageNet 上进行分类训练的主干开始训练的。这在 repo 中的任何地方都没有提到,但我认为是这种情况,因为据我所知,这是标准做法。 .config
中的 configs/tf2
文件的这一部分也暗示了这一点:
train_config: {
fine_tune_checkpoint: "PATH_TO_BE_CONFIGURED/ckpt-0"
fine_tune_checkpoint_version: V2
fine_tune_checkpoint_type: "classification"
.
.
如果情况并非如此,而且我现在对对象检测的方式确实过时了,请有人纠正我。
为了研究目的,我需要在 COCO 的修改版本上重新训练一个检测器(在我的例子中是一个 EfficientDet-Bx)。我想尽可能地遵循原始训练配置进行比较,包括特征提取器初始化。在哪里可以找到这些特征提取器权重?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。