Python OpenCV 高 CPU 占用率问题

用户正在使用 Python OpenCV 开发一个自动驾驶汽车的项目。在项目中，他使用来自 IP 摄像头或视频文件的视频流作为输入，并使用 SVM (libSVM) 对视频流中的图像进行处理。然而，在运行一段时间后，他发现 CPU 占用率会飙升至 100%，并且帧率会下降到 1FPS 以下。他怀疑问题可能出在他的代码中，但无法找到具体原因。

2. 解决方案

经过分析，发现 CPU 占用率飙升的原因是代码中存在一个死循环。在代码中，用户使用 cv.WaitKey(7) 函数来等待用户输入。但是，当用户没有输入任何内容时，cv.WaitKey(7) 函数会一直处于等待状态，导致程序无法继续运行。因此，CPU 占用率会飙升至 100%，并且帧率会下降到 1FPS 以下。

为了解决这个问题，可以将 cv.WaitKey(7) 函数替换为 cv.waitKey(1)。这样，程序就不会陷入死循环，并且 CPU 占用率和帧率都会恢复正常。

以下是修改后的代码：

while True:
    frame = None
    frame = cv.QueryFrame(vidFile)

    cv.ShowImage("selected", frame)
    cv.Smooth(frame, frame, cv.CV_BLUR, 5, 5)

    sub = cv.GetSubRect(frame, selection)

    cv.ShowImage("selected", sub)

    cv.Smooth(sub, sub, cv.CV_BLUR, 5, 5)
    _hsv = cv.CreateImage(cv.GetSize(sub), 8, 3)
    cv.CvtColor(sub, _hsv, cv.CV_BGR2HSV)
    _hue = cv.CreateImage(cv.GetSize(sub), 8, 1)
    cv.Split(_hsv, _hue, None, None, None)

    # Convert to HSV and keep the hue
    hsv = cv.CreateImage(cv.GetSize(frame), 8, 3)
    cv.CvtColor(frame, hsv, cv.CV_BGR2HSV)

    hue = cv.CreateImage(cv.GetSize(frame), 8, 1)

    cv.Split(hsv, hue, None, None, None)

    # Compute back projection
    backproject = cv.CreateImage(cv.GetSize(frame), 8, 1)

    cv.CalcArrBackProject([hue], backproject, hist)

    x,y,w,h = selection

    cv.Rectangle(frame, (x,y), (x+w,y+h), (255,255,255))

    cv.CalcArrHist( [_hue], hist, 0)
    (_, max_val, _, _) = cv.GetMinMaxHistValue(hist)

    threshold=100
    colour=255

    cv.Threshold(backproject,backproject, threshold,colour,cv.CV_THRESH_BINARY)
    cv.Smooth(backproject, backproject, cv.CV_BLUR, 5, 5)
    cv.Dilate(backproject,backproject,None,5)
    cv.Smooth(backproject, backproject, cv.CV_BLUR, 5, 5)
    cv.Dilate(backproject,backproject,None,5)
    cv.Smooth(backproject, backproject, cv.CV_BLUR, 5, 5)
    cv.Dilate(backproject,backproject,None,5)

    cv.Rectangle(backproject, (0,0), (640,280), cv.RGB(0, 0, 0), -1)

    SVMPrediction = predict(backproject)
    moveDirection = directions[SVMPrediction]
    #sendCommandToCarWifi(moveDirection,prevDirection)
    sendCommandToCarSerial(moveDirection,prevDirection)

    #print moveDirection
    cv.ShowImage("Live", frame)
    cv.ShowImage("Backproject", backproject)

    c = cv.WaitKey(1) % 0x100
    if c == 27:
        break

修改后的代码中，cv.WaitKey(1) 函数每隔 1 毫秒检查一次是否有用户输入。如果没有用户输入，程序就会继续运行，不会陷入死循环。因此，CPU 占用率和帧率都会恢复正常。

此外，需要注意的是，在实际应用中，可能还需要对代码进行其他优化，以提高程序的性能。例如，可以使用多线程或多进程来提高程序的并行性，或者可以使用 GPU 来加速图像处理过程。