machine-learning - 将 YOLO 与 vgg 分类器连接
问题描述
我有一个 YOLO 模型来检测交通标志的位置，我有另一个模型来确定标志是什么我的问题是如何将检测到的标志从 YOLO 发送到分类器，然后在边界框上方写上正确的标签，如图像以下
解决方案
您可以为此使用 OpenCV。使用或加载预训练模型（weights、cfg、caffemodel、prototxt、onnx）。readNetreadNetFrom...
对于视频，用于OpenCV dnn blobFromImage将每一帧传递给检测边界框的 yolo 网络。您可能需要进行非最大抑制。根据置信度裁剪边界框，然后再次将裁剪后的图像传递dnn给应该给出分类结果的 vgg 网络。
要编写标签，请获取(x,y)每个边界框的位置，并使用putText字体大小作为原始输入图像帧上的偏移量。