Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/dltrack.py @ 1278:8e61ff3cd503

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
correct bug to take into account first frame num in config, and other related bugs in dltrack.py
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 27 Jun 2024 15:31:36 -0400
parents b2f90cada58f
children 2abeccdbb985
comparison
equal deleted inserted replaced
1277:7493751bfe19 1278:8e61ff3cd503
30 parser.add_argument('--undistort', dest = 'undistort', help = 'undistort the video', action = 'store_true') 30 parser.add_argument('--undistort', dest = 'undistort', help = 'undistort the video', action = 'store_true')
31 parser.add_argument('--intrinsic', dest = 'intrinsicCameraMatrixFilename', help = 'name of the intrinsic camera file') 31 parser.add_argument('--intrinsic', dest = 'intrinsicCameraMatrixFilename', help = 'name of the intrinsic camera file')
32 parser.add_argument('--distortion-coefficients', dest = 'distortionCoefficients', help = 'distortion coefficients', nargs = '*', type = float) 32 parser.add_argument('--distortion-coefficients', dest = 'distortionCoefficients', help = 'distortion coefficients', nargs = '*', type = float)
33 parser.add_argument('--display', dest = 'display', help = 'show the raw detection and tracking results', action = 'store_true') 33 parser.add_argument('--display', dest = 'display', help = 'show the raw detection and tracking results', action = 'store_true')
34 parser.add_argument('--no-image-coordinates', dest = 'notSavingImageCoordinates', help = 'not saving the raw detection and tracking results', action = 'store_true') 34 parser.add_argument('--no-image-coordinates', dest = 'notSavingImageCoordinates', help = 'not saving the raw detection and tracking results', action = 'store_true')
35 parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to process', type = int, default = 0) 35 parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to process', type = int)
36 parser.add_argument('-l', dest = 'lastFrameNum', help = 'number of last frame number to process', type = int, default = inf) 36 parser.add_argument('-l', dest = 'lastFrameNum', help = 'number of last frame number to process', type = int, default = inf)
37 parser.add_argument('--conf', dest = 'confidence', help = 'object confidence threshold for detection', type = float, default = 0.25) 37 parser.add_argument('--conf', dest = 'confidence', help = 'object confidence threshold for detection', type = float, default = 0.25)
38 parser.add_argument('--bike-prop', dest = 'bikeProportion', help = 'minimum proportion of time a person classified as bike or motorbike to be classified as cyclist', type = float, default = 0.2) 38 parser.add_argument('--bike-prop', dest = 'bikeProportion', help = 'minimum proportion of time a person classified as bike or motorbike to be classified as cyclist', type = float, default = 0.2)
39 parser.add_argument('--cyclist-iou', dest = 'cyclistIou', help = 'IoU threshold to associate a bike and ped bounding box', type = float, default = 0.15) 39 parser.add_argument('--cyclist-iou', dest = 'cyclistIou', help = 'IoU threshold to associate a bike and ped bounding box', type = float, default = 0.15)
40 parser.add_argument('--cyclist-match-prop', dest = 'cyclistMatchingProportion', help = 'minimum proportion of time a bike exists and is associated with a pedestrian to be merged as cyclist', type = float, default = 0.3) 40 parser.add_argument('--cyclist-match-prop', dest = 'cyclistMatchingProportion', help = 'minimum proportion of time a bike exists and is associated with a pedestrian to be merged as cyclist', type = float, default = 0.3)
41 #parser.add_argument('--max-temp-overal', dest = 'maxTemporalOverlap', help = 'maximum proportion of time to merge 2 bikes associated with same pedestrian', type = float, default = 0.05) 41 #parser.add_argument('--max-temp-overal', dest = 'maxTemporalOverlap', help = 'maximum proportion of time to merge 2 bikes associated with same pedestrian', type = float, default = 0.05)
42 42
43 args = parser.parse_args() 43 args = parser.parse_args()
44 params, videoFilename, databaseFilename, homography, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) 44 params, videoFilename, databaseFilename, homography, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args)
45
46 print(params, videoFilename, databaseFilename, homography)
45 47
46 if args.homographyFilename is not None: 48 if args.homographyFilename is not None:
47 homography = np.loadtxt(args.homographyFilename) 49 homography = np.loadtxt(args.homographyFilename)
48 if args.intrinsicCameraMatrixFilename is not None: 50 if args.intrinsicCameraMatrixFilename is not None:
49 intrinsicCameraMatrix = np.loadtxt(args.intrinsicCameraMatrixFilename) 51 intrinsicCameraMatrix = np.loadtxt(args.intrinsicCameraMatrixFilename)
54 if args.lastFrameNum is not None: 56 if args.lastFrameNum is not None:
55 lastFrameNum = args.lastFrameNum 57 lastFrameNum = args.lastFrameNum
56 elif args.configFilename is not None: 58 elif args.configFilename is not None:
57 lastFrameNum = params.lastFrameNum 59 lastFrameNum = params.lastFrameNum
58 else: 60 else:
59 lastFrameNum = args.lastFrameNum 61 lastFrameNum = np.inf
60 if args.maskFilename is not None: 62 if args.maskFilename is not None:
61 mask = cv2.imread(args.maskFilename, cv2.IMREAD_GRAYSCALE) 63 mask = cv2.imread(args.maskFilename, cv2.IMREAD_GRAYSCALE)
62 elif params is not None and params.maskFilename is not None: 64 elif params is not None and params.maskFilename is not None:
63 mask = cv2.imread(params.maskFilename, cv2.IMREAD_GRAYSCALE) 65 mask = cv2.imread(params.maskFilename, cv2.IMREAD_GRAYSCALE)
64 else: 66 else:
93 frameNum = firstFrameNum 95 frameNum = firstFrameNum
94 capture.set(cv2.CAP_PROP_POS_FRAMES, frameNum) 96 capture.set(cv2.CAP_PROP_POS_FRAMES, frameNum)
95 97
96 success, frame = capture.read() 98 success, frame = capture.read()
97 if not success: 99 if not success:
98 print('Input {} could not be read. Exiting'.format(args.videoFilename)) 100 print('Input {} could not be read. Exiting'.format(videoFilename))
99 import sys; sys.exit() 101 import sys; sys.exit()
100 102
101 results = model.track(source=frame, tracker=args.trackerFilename, classes=list(moving.cocoTypeNames.keys()), conf=args.confidence, persist=True, verbose=False) 103 results = model.track(source=frame, tracker=args.trackerFilename, classes=list(moving.cocoTypeNames.keys()), conf=args.confidence, persist=True, verbose=False)
102 while capture.isOpened() and success and frameNum <= lastFrameNum: 104 while capture.isOpened() and success and frameNum <= lastFrameNum:
103 result = results[0] 105 result = results[0]
235 if f.length() != len(f.tmpPositions): # interpolate 237 if f.length() != len(f.tmpPositions): # interpolate
236 f.positions = moving.Trajectory.fromPointDict(f.tmpPositions) 238 f.positions = moving.Trajectory.fromPointDict(f.tmpPositions)
237 else: 239 else:
238 f.positions = moving.Trajectory.fromPointList(list(f.tmpPositions.values())) 240 f.positions = moving.Trajectory.fromPointList(list(f.tmpPositions.values()))
239 if not args.notSavingImageCoordinates: 241 if not args.notSavingImageCoordinates:
240 storage.saveTrajectoriesToSqlite(utils.removeExtension(args.databaseFilename)+'-bb.sqlite', list(objects.values()), 'object') 242 storage.saveTrajectoriesToSqlite(utils.removeExtension(databaseFilename)+'-bb.sqlite', list(objects.values()), 'object')
241 # project and smooth 243 # project and smooth
242 for num, obj in objects.items(): 244 for num, obj in objects.items():
243 features = obj.getFeatures() 245 features = obj.getFeatures()
244 # possible to save bottom pedestrians? not consistent with other users 246 # possible to save bottom pedestrians? not consistent with other users
245 # if moving.userTypeNames[obj.getUserType()] == 'pedestrian': 247 # if moving.userTypeNames[obj.getUserType()] == 'pedestrian':
261 feature.smoothPositions(smoothingHalfWidth, replace = True)#f.positions = f.getPositions().filterMovingWindow(smoothingHalfWidth) 263 feature.smoothPositions(smoothingHalfWidth, replace = True)#f.positions = f.getPositions().filterMovingWindow(smoothingHalfWidth)
262 feature.computeVelocities() 264 feature.computeVelocities()
263 obj.features=[feature] 265 obj.features=[feature]
264 obj.featureNumbers = [featureNum] 266 obj.featureNumbers = [featureNum]
265 #saving 267 #saving
266 storage.saveTrajectoriesToSqlite(args.databaseFilename, list(objects.values()), 'object') 268 storage.saveTrajectoriesToSqlite(databaseFilename, list(objects.values()), 'object')
267 269
268 270
269 271
270 # todo save bbox and mask to study localization / representation 272 # todo save bbox and mask to study localization / representation
271 # apply quality checks deviation and acceleration bounds? 273 # apply quality checks deviation and acceleration bounds?