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