我使用 yolo 进行汽车检测,还训练了另一个 yolo 模型进行车牌检测,该模型可以检测所有车辆的车牌。我想加入这两个仅检测汽车牌照的代码。上图也检测公共汽车、卡车的车牌。有什么方法可以仅当检测到的车辆是汽车时才能检测到车牌?
您可以...将原始图像提交给汽车检测模型。 NMS 输出将包含类别、置信度和边界框坐标。对于等于 car 的对象类,使用 OpenCV 根据缩放后的边界框输出裁剪原始图像,并将新图像提交给车牌模型。第二个模型的输出应仅包含汽车区域内的车牌。
问题解决了吗?