python - 如何在Python中使用OpenCV在检测部分绘制矩形？

Question

我创建了一个将normal_car与cop_car进行分类的模型。

如果我将图像传递给model，它将提供正确的输出。

但是，当图像中有很多对象时，如何在
这两个(标签:normal_Car，police_Car)？

我可以在每个对象的图像中绘制一个矩形(但无需使用模型)。

怎么做？

Answer 1

当您在图像中找到对象时，OpenCV会为您提供其位置。您将裁剪并通过预训练的模型进行预测。

您可以加载H5文件模型或冻结的文件。

例如:

model = load_model("modelname.h5")

在图像中查找对象并提取对象位置(x，y，w，h)

(_, Contours ,_) = cv2.findContours(Image,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
for area in Contours:
    (x, y, w, h) = cv2.boundingRect(area)
    ObjROI = Image[y : y+h, x : x+w]
    p = model.predict(cv2.resize(ObjROI, (<size of CNN input>)))

现在，您可以精确找到所找到的对象及其位置，绘制矩形并将其放置为文本。