comparison scripts/dltrack.py @ 1268:27b206d118b7

bug at the end of video
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 31 May 2024 16:52:51 -0400
parents 0f5bebd62a55
children 20a5e1292321
comparison
equal deleted inserted replaced
1267:ad60e5adf084 1268:27b206d118b7
132 key = cv2.waitKey() 132 key = cv2.waitKey()
133 if cvutils.quitKey(key): 133 if cvutils.quitKey(key):
134 break 134 break
135 frameNum += 1 135 frameNum += 1
136 success, frame = capture.read() 136 success, frame = capture.read()
137 results = model.track(source=frame, persist=True) 137 if success:
138 results = model.track(source=frame, persist=True)
138 capture.release() 139 capture.release()
139 cv2.destroyAllWindows() 140 cv2.destroyAllWindows()
140 141
141 # classification 142 # classification
142 shortObjectNumbers = [] 143 shortObjectNumbers = []