Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/display-synced-trajectories.py @ 836:7058a40a4bbc

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
updated metadata and code to merge features from different cameras
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 06 Jul 2016 10:18:35 -0400
parents 02f2809c2f66
children e01cabca4c55
comparison
equal deleted inserted replaced
835:f3ae72d86762 836:7058a40a4bbc
26 sys.exit() 26 sys.exit()
27 27
28 dirname = os.path.split(args.metadataFilename)[0] 28 dirname = os.path.split(args.metadataFilename)[0]
29 29
30 startTime = datetime.strptime(args.startTime, utils.datetimeFormat) 30 startTime = datetime.strptime(args.startTime, utils.datetimeFormat)
31 # TODO issue with framerate
32 if startTime > mergedSequence.startTime: 31 if startTime > mergedSequence.startTime:
33 mergedFirstFrameNum = (startTime-mergedSequence.startTime).seconds* 32 mergedFirstFrameNum = (startTime-mergedSequence.startTime).seconds*mergedSequence.cameraView.cameraType.frameRate
34 33
35 videoSequences = session.query(VideoSequence).filter(VideoSequence.site == mergedSequence.site).filter(VideoSequence.startTime <= startTime).all() 34 cameraViews = session.query(CameraView).filter(CameraView.site == mergedSequence.site).filter(CameraView.virtual == False)
36 videoSequences.remove(mergedSequence) 35 videoSequences = session.query(VideoSequence).filter(VideoSequence.virtual == False).filter(VideoSequence.startTime <= startTime).all()
37 videoSequences = [v for v in videoSequences if v.containsInstant(startTime)] 36 #videoSequences.remove(mergedSequence)
37 videoSequences = [v for v in videoSequences if v.cameraView in cameraViews and v.containsInstant(startTime)]
38 filenames = [dirname+os.path.sep+v.getVideoSequenceFilename() for v in videoSequences] 38 filenames = [dirname+os.path.sep+v.getVideoSequenceFilename() for v in videoSequences]
39 firstFrameNums = [v.getFrameNum(startTime) for v in videoSequences] 39 firstFrameNums = [v.getFrameNum(startTime) for v in videoSequences]
40 windowNames = [v.cameraView.description for v in videoSequences] 40 windowNames = [v.cameraView.description for v in videoSequences]
41 41
42 objects = storage.loadTrajectoriesFromSqlite(dirname+os.path.sep+mergedSequence.getDatabaseFilename(), args.trajectoryType) 42 objects = storage.loadTrajectoriesFromSqlite(dirname+os.path.sep+mergedSequence.getDatabaseFilename(), args.trajectoryType)