python - 使用opencv python在灰度视频中绘制彩色矩形

Question

这是我的代码
我需要在灰色视频流上绘制一条颜色线和一个矩形。
在我的代码中有一些错误，因为我的线和矩形是黑色的，但不是。

import cv2 
 
cap = cv2.VideoCapture(0) 

if (cap.isOpened() == False): 
    print("Unable to read camera feed")

frame_width = int(cap.get(3))
frame_height = int(cap.get(4))

out = cv2.VideoWriter('output.avi',cv2.VideoWriter_fourcc('M','J','P','G'), 10, (frame_width,frame_height),0)

while(cap.isOpened()): 
    
    ret, frame = cap.read() 
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    
    # draw line
    start_point = (0, 0)
    end_point = (250, 250)
    color = (0, 255, 0)
    thickness = 5
    gray = cv2.line(img=gray, pt1=start_point, pt2=end_point, color=color, thickness=thickness, lineType=8, shift=0)
    
    # draw rectangle
    x1,y1 = 200, 200
    x2,y2 = 250, 250  
    gray = cv2.rectangle(gray,(x1, y1), (x2, y2),color, 2)
    
    cv2.imshow('webcam(1)', gray)

    out.write(gray)
    if cv2.waitKey(1) & 0xFF == ord('q'): 
        break
  

cap.release() 
out.release()
cv2.destroyAllWindows() 

Answer 1

要绘制颜色元素，您必须将图像转换回BGR

 gray_BGR = cv2.cvtColor(gray cv2.COLOR_GRAY2BGR)

转换为灰色不仅可以将颜色转换为灰色，而且还可以将每个像素从三个值(B，G，R)减少为只能保留灰色的单个值。
如果检查 frame.shape和 gray.shape，则会发现区别。
首先将具有 (height, width, 3)，其次将仅具有 (height, width)，这意味着 (height, width, 1)