drobne poprawki

This commit is contained in:
Krzysztof Strzelecki 2020-02-15 00:24:35 +01:00
parent 50a90d23d6
commit 999559aa6a

16
main.py
View File

@ -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)