Mercurial Hosting > traffic-intelligence
diff scripts/safety-analysis.py @ 1252:fe35473acee3
adding method to compute PET using polygon for the outline of a vehicle (bird eye view of the vehicle)
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Fri, 22 Mar 2024 14:33:25 -0400 |
| parents | 14140b55e580 |
| children |
line wrap: on
line diff
--- a/scripts/safety-analysis.py Fri Mar 15 17:05:54 2024 -0400 +++ b/scripts/safety-analysis.py Fri Mar 22 14:33:25 2024 -0400 @@ -90,11 +90,11 @@ interactions = events.createInteractions(objects) if args.nProcesses == 1: - processed = events.computeIndicators(interactions, not args.noMotionPrediction, args.computePET, predictionParameters, params.collisionDistance, params.predictionTimeHorizon, params.crossingZones, False, None) + processed = events.computeIndicators(interactions, not args.noMotionPrediction, args.computePET, predictionParameters, params.collisionDistance, False, params.predictionTimeHorizon, params.crossingZones, False, None) else: pool = Pool(processes = args.nProcesses) nInteractionPerProcess = int(np.ceil(len(interactions)/float(args.nProcesses))) - jobs = [pool.apply_async(events.computeIndicators, args = (interactions[i*nInteractionPerProcess:(i+1)*nInteractionPerProcess], not args.noMotionPrediction, args.computePET, predictionParameters, params.collisionDistance, params.predictionTimeHorizon, params.crossingZones, False, None)) for i in range(args.nProcesses)] + jobs = [pool.apply_async(events.computeIndicators, args = (interactions[i*nInteractionPerProcess:(i+1)*nInteractionPerProcess], not args.noMotionPrediction, args.computePET, predictionParameters, params.collisionDistance, False, params.predictionTimeHorizon, params.crossingZones, False, None)) for i in range(args.nProcesses)] processed = [] for job in jobs: processed += job.get()