我正在Keras中使用Yolo v3的以下实现,通过安装在面向铁路的铁路前方的摄像机来检测铁路轨道上的物体:
https://github.com/qqwweee/keras-yolo3
我有一个带有标签对象的数据集,它们与铁路线的距离以米为单位。我想使用冻结的yolo权重,并通过附加输出(即检测到的物体的距离)来训练网络。
我在python 3.5上使用Keras 2.1.5。所以我的问题是,我是否需要在体系结构中逐层走,以查看我从哪里获取输出,或者只是将其添加到最后一层。任何帮助都会很棒。