使用Haarcascade检测到的面部，如何在不同颜色的相同面上绘制边界框

Question

正如我在主题上所说的，下面的代码使用haarcascade-Opencv / Python检测图像中的每个面。

代码检测所有面孔，

但我需要在图像中检测相同的面部，然后绘制不同颜色的边界框

Iam beginer我用谷歌搜索我怎么能做到这一点，但我不够。

我知道我需要数据集，但我不知道如何训练它以及如何实现下面的代码..

有没有人有过这方面的经验？

也许有人可以根据下面的代码给我一个例子然后我会尝试按照他的步骤。

检测面部的代码：

import cv2
import matplotlib.pyplot as plt
test_image = cv2.imread("C:\Users\erdal.alimovski\Desktop\faces.jpg")

test_image_gray = cv2.cvtColor(test_image, cv2.COLOR_BGR2GRAY)
plt.imshow(test_image_gray, cmap='gray')

def convertToRGB(image):
return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

haar_cascade_face = cv2.CascadeClassifier('C:\Users\haarcascade_frontalface_default.xml')

faces_rects = haar_cascade_face.detectMultiScale(test_image_gray, scaleFactor = 1.2, minNeighbors = 5);

print('Faces found: ', len(faces_rects))

for (x,y,w,h) in faces_rects:
  cv2.rectangle(test_image, (x, y), (x+w, y+h), (0, 255, 0), 2)

cv2.imshow("yeni",test_image)
cv2.waitKey(10000)

Answer 1

在cv2.rectangle(test_image, (x, y), (x+w, y+h), (0, 255, 0), 2)行，参数（0,255,0）表示每次都绘制一个绿色矩形（因为255是绿色部分）。

你可以

1-在此参数上传递随机生成的颜色

2-初始化你想要使用的颜色列表并在for (x,y,w,h) in faces_rects:中循环它

colorList=[(0,255,0),(255,0,0),(0,0,255)] # etc ...
i=0
for (x,y,w,h) in faces_rects:
  cv2.rectangle(test_image, (x, y), (x+w, y+h), colorList[i], 2)
  i=i+1