drobne poprawki i dodatkowy plik video

This commit is contained in:
Krzysztof Strzelecki 2020-02-15 02:46:14 +01:00
parent 999559aa6a
commit ed451efcec
3 changed files with 16 additions and 9 deletions

BIN
data/video_3.mp4 Normal file

Binary file not shown.

17
main.py
View File

@ -7,7 +7,7 @@ counter_up=0 #licznik pojazdow jadacych w gore
counter_down=0 #licznik pojazdow jadacych w dol counter_down=0 #licznik pojazdow jadacych w dol
while True: while True:
user_choice = input("Wybierz nagranie:\n1. video_1.mp4\n2. video_2.avi\n") user_choice = input("Wybierz nagranie:\n1. video_1.mp4\n2. video_2.avi\n3. video_3.mp4\n")
if user_choice == "1": if user_choice == "1":
input_video = 'data/video_1.avi' input_video = 'data/video_1.avi'
vid_name = 'video_1' vid_name = 'video_1'
@ -18,6 +18,13 @@ while True:
vid_name = 'video_2' vid_name = 'video_2'
kernelSize = 3 kernelSize = 3
break break
elif user_choice == "3":
input_video = 'data/video_3.mp4'
vid_name = 'video_3'
kernelSize = 3
break
cap = cv2.VideoCapture(input_video) cap = cv2.VideoCapture(input_video)
@ -133,10 +140,10 @@ while(cap.isOpened()):
new = False new = False
i.updateCoords(cx, cy) i.updateCoords(cx, cy)
#sprawdz czy prekroczyl mete #sprawdz czy prekroczyl mete
if i.going_UP(lineLower,lineUpper) == True: if i.going_UP(lineUpper) == True:
counter_up+=1 counter_up+=1
print("Pojazd o ID:",i.getId(),'przekroczyl linie mety w gore o czasie:', time.strftime("%c")) print("Pojazd o ID:",i.getId(),'przekroczyl linie mety w gore o czasie:', time.strftime("%c"))
elif i.going_DOWN(lineLower,lineUpper) == True: elif i.going_DOWN(lineLower) == True:
counter_down+=1 counter_down+=1
print("Pojazd o ID:", i.getId(), 'przekroczyl linie mety w dol o czasie:', time.strftime("%c")) print("Pojazd o ID:", i.getId(), 'przekroczyl linie mety w dol o czasie:', time.strftime("%c"))
break break
@ -174,10 +181,10 @@ while(cap.isOpened()):
cv2.putText(frame, str_up, (10, 40), font, 0.4, lineUpper_color, 1, cv2.LINE_AA) 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, (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) 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 #frame = cv2.resize(frame, (0, 0), None, size, size) # rozmiar obrazu x2
cv2.imshow('Frame',frame) cv2.imshow('Frame',frame)
key = cv2.waitKey(60) key = cv2.waitKey(30)
if key == 27: # Wyjdz po nacisnieciu escape if key == 27: # Wyjdz po nacisnieciu escape
break break

View File

@ -48,10 +48,10 @@ class Car:
def timedOut(self): def timedOut(self):
return self.done return self.done
def going_UP(self, mid_start, mid_end): def going_UP(self, finishLine):
if len(self.tracks)>=2: if len(self.tracks)>=2:
if self.state=='0': if self.state=='0':
if self.tracks[-1][1]<mid_end and self.tracks[-2][1]>=mid_end: if self.tracks[-1][1]<finishLine and self.tracks[-2][1]>=finishLine:
self.state='1' self.state='1'
self.dir='up' self.dir='up'
return True return True
@ -62,10 +62,10 @@ class Car:
else: else:
return False return False
def going_DOWN(self,mid_start,mid_end): def going_DOWN(self,finishLine):
if len(self.tracks)>=2: if len(self.tracks)>=2:
if self.state=='0': if self.state=='0':
if self.tracks[-1][1]>mid_start and self.tracks[-2][1]<=mid_start: if self.tracks[-1][1]>finishLine and self.tracks[-2][1]<=finishLine:
self.state='1' self.state='1'
self.dir='down' self.dir='down'
return True return True