通过使用Faster-RCNN的预训练模型将其用作特征提取器

时间:2019-07-20 13:12:32

标签: python tensorflow object-detection transfer-learning faster-rcnn

我正在尝试使用https://github.com/tensorflow/models/tree/master/research/object_detection上的Faster-RCNN-Inception-v2预训练模型 进行转移学习。我的项目必须加载此模型,省略FC层,并将其余部分用作特征提取器,以将每张图片转换为一组特征。 我有几个问题。首先,它是用tensorflow核心编写的,我几乎不理解,因为我已经用keras编写了该项目的其余部分。幸运的是,我阅读了教程页面,并找到了示例代码来加载和使用该模型。 因为我想省略FC层,所以我必须知道如何删除最后一层并从适当的层获取输出。但是,我找不到图层名称,也找不到任何描述图层名称的方式的手册或教程。我使用张量板查看模型,但是它是如此复杂,以至于找不到输入/输出。 同样在文件夹/ research / object_detection / models /中,有一些文件名为例如fast_rcnn_inception_v2_feature_extractor_test.py,它似乎已经实现了我想要的功能。但是它没有注释或教程,因此我可以知道自己掌握的代码。

如果您可以指导我在哪里阅读(任何教程),或者有简单的解决方案,请指导我完成,我非常感谢。 我很抱歉,因为我是新手,希望转学。请帮忙:-)

0 个答案:

没有答案