diff --git a/main.py b/main.py index b504996..f796228 100644 --- a/main.py +++ b/main.py @@ -68,7 +68,7 @@ points_downLimit= np.array([point7,point8], np.int32) sub=cv2.createBackgroundSubtractorMOG2() # utworz background subtractor -size = 1.0 #wielkosc obrazu +size = 2.0 #wielkosc obrazu cars = [] maxAllowedAge = 3 #przez ile klatek "zgubiony" pojazd bedzie pozostawal na liscie do "sledzenia" id = 1 @@ -81,12 +81,11 @@ while(cap.isOpened()): if ret == True: #przetwarzanie kadru - #frame = cv2.resize(frame, (0, 0), None, size, size) # rozmiar obrazu. Bez zmian #cv2.imshow("bez zmian", frame) # wyswietlanie filmu wejsciowego bez zmian #gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) #cv2.imshow("czarnobialy", gray) # wyswietlanie filmu w skali szarosci fgMask = sub.apply(frame) # uzycie background subtraction - fgMask2 = sub.apply(frame) + #fgMask2 = sub.apply(frame) #cv2.imshow("fgMask", fgMask) # operacje morfologiczne. Wg tutoriala na docs.opencv.org. Opening, closing - usuwanie szumów #Binarization @@ -103,7 +102,7 @@ while(cap.isOpened()): #Closing is reverse of Opening, Dilation followed by Erosion. #It is useful in closing small holes inside the foreground objects, or small black points on the object. mask = cv2.morphologyEx(mask,cv2.MORPH_CLOSE,kernelClosing) - cv2.imshow("closing", mask) + #cv2.imshow("closing", mask) # kontury (obrysowywanie) contours,hierarchy=cv2.findContours(mask,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE) # RETR_EXTERNAL - bierz pod uwage najbardziej 'zewnetrzne' kontury @@ -171,10 +170,11 @@ while(cap.isOpened()): frame = cv2.polylines(frame,[points_LineUpper],False,lineUpper_color,thickness=2) frame = cv2.polylines(frame,[points_upLimit],False,(255,255,255),thickness=1) frame = cv2.polylines(frame,[points_downLimit],False,(255,255,255),thickness=1) - cv2.putText(frame, str_up, (10, 40), font, 0.5, (255, 255, 255), 2, cv2.LINE_AA) #biale tlo zeby licznik byl wyrazniejszy - cv2.putText(frame, str_up, (10, 40), font, 0.5, lineUpper_color, 1, cv2.LINE_AA) - cv2.putText(frame, str_down, (10, 90), font, 0.5, (255, 255, 255), 2, cv2.LINE_AA) #biale tlo zeby licznik byl wyrazniejszy - cv2.putText(frame, str_down, (10, 90), font, 0.5, lineLower_color, 1, cv2.LINE_AA) + cv2.putText(frame, str_up, (10, 40), font, 0.4, (255, 255, 255), 2, cv2.LINE_AA) #biale tlo zeby licznik byl wyrazniejszy + cv2.putText(frame, str_up, (10, 40), font, 0.4, lineUpper_color, 1, cv2.LINE_AA) + cv2.putText(frame, str_down, (10, 90), font, 0.4, (255, 255, 255), 2, cv2.LINE_AA) #biale tlo zeby licznik byl wyrazniejszy + cv2.putText(frame, str_down, (10, 90), font, 0.4, lineLower_color, 1, cv2.LINE_AA) + frame = cv2.resize(frame, (0, 0), None, size, size) # rozmiar obrazu x2 cv2.imshow('Frame',frame) key = cv2.waitKey(60)