向yolo v3添加另一个输出神经元

时间:2019-05-21 16:04:05

标签: python-3.x machine-learning keras deep-learning yolo

我正在Keras中使用Yolo v3的以下实现,通过安装在面向铁路的铁路前方的摄像机来检测铁路轨道上的物体:

https://github.com/qqwweee/keras-yolo3

我有一个带有标签对象的数据集,它们与铁路线的距离以米为单位。我想使用冻结的yolo权重,并通过附加输出(即检测到的物体的距离)来训练网络。

我在python 3.5上使用Keras 2.1.5。所以我的问题是,我是否需要在体系结构中逐层走,以查看我从哪里获取输出,或者只是将其添加到最后一层。任何帮助都会很棒。

0 个答案:

没有答案