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

Question

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

代码检测所有面孔，

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

我是初学者，我在谷歌上搜索如何做到这一点，但我做得不够。

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

以前有人有过这方面的经验吗？

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

检测人脸的代码:

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