使用keras在yolo上转移学习

时间:2019-06-23 09:32:15

标签: keras deep-learning conv-neural-network yolo transfer-learning

我正在研究对象检测用例。我得到了需要在视频中检测到的徽标图像。我正在喀拉拉邦这样做。我按照以下博客将yolo权重转换为keras模型。 https://forum.drawbot.com/topic/26/using-a-virtualenv-in-drawbot/7)...this

现在我想用我的数据集训练该keras模型(转移学习)。我得到了正确的xmin,ymin,xmax和ymax数据集。

我的方法:删除最后一层,并使用sigmoid函数定义类。但在上面的链接中说,这个keras模型可以预测边界框和类别。我不确定如何从模型本身获得这些边界框。

0 个答案:

没有答案