2017-05-31 3 views
0

Ich habe ein Video, das durch Hintergrundsubtraktion auf einem anderen Video erhalten wurde. Jetzt muss ich die Blob-Erkennung in diesem Video durchführen und die Blobs mit einem roten Rand markieren. Mein Code ist wie folgt:Blob-Erkennung im Hintergrund nicht subtrahiert video opencv python

capture = cv2.VideoCapture('bw.avi') 
size = (int(capture.get(cv2.CAP_PROP_FRAME_WIDTH)), 
int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT))) 
fourcc = cv2.VideoWriter_fourcc(*'DIVX') 
video = cv2.VideoWriter('harsha_blob.avi', fourcc, 10.0,size) 

while (1): 
    ret, im = capture.read() 
    im = cv2.convertScaleAbs(im) 

    params = cv2.SimpleBlobDetector_Params() 
    params.blobColor = 0 
    params.filterByColor = True 
    params.minArea = 0 
    params.filterByArea = False 
    params.minThreshold = 120; 
    params.maxThreshold = 255; 
    detector = cv2.SimpleBlobDetector_create(params) 

    keypoints = detector.detect(im) 

    # Draw detected blobs as red circles. 
    # cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS ensures the size of the circle corresponds to the size of blob 
    im_with_keypoints = cv2.drawKeypoints(im, keypoints, np.array([]), (0,0,255), cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS) 


    if ret==True: 

     video.write(im_with_keypoints) 


    else: 
     capture.release() 
     video.release() 
     break 

    k = cv2.waitKey(30) & 0xff 
    if k == 27: 
     break 

cv2.destroyAllWindows() 

Der Hintergrund subtrahiert Video invertiert wurde die Blobs schwarz und der Hintergrund weiß zu machen, wie Blob Erkennung schwarz/grau Blobs findet. Ich bin in der Lage, die Blobs in einem einzelnen Frame zu erkennen, erhalte jedoch den folgenden Fehler, wenn ich versuche, es auf dem Video auszuführen.

im_with_keypoints = cv2.drawKeypoints(im, keypoints, np.array([]), (0,0,255), cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS) 

error: C:\build\master_winpack-bindings-win64-vc14-static\opencv\modules\features2d\src\draw.cpp:115: error: (-215) !outImage.empty() in function cv::drawKeypoints 

Warum erhalte ich den Fehler? Und wie repariere ich das?

Antwort

1

!outImage.empty() geschieht am Ende des Videos, wenn es keine weiteren Rahmen und ret, im = capture.read() Erträge sind ret==False. Sie sollten diese Bedingung überprüfen, bevor Sie die Schlüsselpunkte für die Blobs finden.

while (1): 
    ret, im = capture.read() 
    if not ret: 
     break 

    # blob detection code 
Verwandte Themen