如何在不丢失检测盒的情况下进行转移学习

时间:2019-10-21 18:46:35

标签: python tensorflow computer-vision object-detection transfer-learning

我正在ImagenetV2对象检测器上进行转移学习。当我使用预训练模型时,model.predict()返回包含对象检测框坐标的结果。转移学习后,我只能得到一个数字输出,从零到一。下面是我的伪代码。

flatten last layer of pretrained model
add layer with 128 nodes
add layer with 1 node, activation sigmoid

我想知道如何进行转移学习,以便我的模型在图像上找到对象并放心返回边界框坐标

0 个答案:

没有答案