Types & classes — github.com/Rex-sys-hk/PlanScope

CodeHub MCPcopy Create free account

hub / github.com/Rex-sys-hk/PlanScope / types & classes

Types & classes70 in github.com/Rex-sys-hk/PlanScope

⨍ Functions 459◇ Types & classes 70

↓ 21 callersClassMLPLayer

src/models/pluto/layers/mlp_layer.py:4

↓ 6 callersClassFourierEmbedding

src/models/pluto/layers/fourier_embedding.py:21

↓ 6 callersClassPlutoFeature

src/features/pluto_feature.py:17

↓ 6 callersClassScopeFeature

src/features/scope_feature.py:17

↓ 5 callersClassOccupancyMap

src/scenario_manager/occupancy_map.py:21

↓ 5 callersClassScenarioManager

src/scenario_manager/scenario_manager.py:47

↓ 3 callersClassBreadthFirstSearchRoadBlock

A class that performs iterative breadth first search. The class operates on the roadblock graph.

src/scenario_manager/utils/bfs_roadblock.py:8

↓ 3 callersClassPointsEncoder

src/models/pluto/layers/embedding.py:254

↓ 2 callersClassAgentEncoder

src/models/pluto/modules/agent_encoder.py:8

↓ 2 callersClassAgentPredictor

src/models/pluto/modules/agent_predictor.py:7

↓ 2 callersClassCollisionChecker

src/utils/collision_checker.py:8

↓ 2 callersClassCostMapManager

src/scenario_manager/cost_map_manager.py:18

↓ 2 callersClassDijkstra

A class that performs dijkstra's shortest path. The class operates on lane level graph search. The goal condition is specified to be if the l

src/scenario_manager/utils/dijkstra.py:9

↓ 2 callersClassESDFCollisionLoss

src/models/pluto/loss/esdf_collision_loss.py:9

↓ 2 callersClassEmergencyBrake

src/post_processing/emergency_brake.py:24

↓ 2 callersClassForwardSimulator

src/post_processing/forward_simulation/forward_simulator.py:13

↓ 2 callersClassMR

src/metrics/mr.py:7

↓ 2 callersClassMapEncoder

src/models/pluto/modules/map_encoder.py:8

↓ 2 callersClassNuplanScenarioRender

src/feature_builders/nuplan_scenario_render.py:41

↓ 2 callersClassPlutoFeatureBuilder

src/feature_builders/pluto_feature_builder.py:38

↓ 2 callersClassPredAvgADE

src/metrics/prediction_avg_ade.py:8

↓ 2 callersClassPredAvgFDE

src/metrics/prediction_avg_fde.py:8

↓ 2 callersClassStaticObjectsEncoder

src/models/pluto/modules/static_objects_encoder.py:8

↓ 2 callersClassTrajectoryEvaluator

src/post_processing/trajectory_evaluator.py:49

↓ 2 callersClassTransformerEncoderLayer

src/models/pluto/layers/transformer.py:40

↓ 2 callersClassWarmupCosLR

src/optim/warmup_cos_lr.py:6

↓ 2 callersClassminADE

Minimum Average Displacement Error minADE: The average L2 distance between the best forecasted trajectory and the ground truth. The be

src/metrics/min_ade.py:9

↓ 2 callersClassminFDE

src/metrics/min_fde.py:9

↓ 1 callersClassBatchKinematicBicycleModel

A batch-wise operating class describing the kinematic motion model where the rear axle is the point of reference.

src/post_processing/forward_simulation/batch_kinematic_bicycle.py:34

↓ 1 callersClassBatchLQRTracker

Implements an LQR tracker for a kinematic bicycle model. Tracker operates on a batch of proposals. Implementation directly based on the nupl

src/post_processing/forward_simulation/batch_lqr.py:38

↓ 1 callersClassConvDownsampler

src/models/pluto/layers/embedding.py:107

↓ 1 callersClassConvTokenizer

src/models/pluto/layers/embedding.py:90

↓ 1 callersClassCustomDataModule

Datamodule wrapping all preparation and dataset creation functionality.

src/custom_training/custom_datamodule.py:103

↓ 1 callersClassDecoderLayer

src/models/pluto/modules/planning_decoder.py:12

↓ 1 callersClassDecoderLayer

src/models/pluto/modules/hierachical_decoder.py:12

↓ 1 callersClassHierachicalDecoder

src/models/pluto/modules/hierachical_decoder.py:89

↓ 1 callersClassMVNLoss

Multiple Scope Average Displacement Error mulADE: The average L2 distance between the best forecasted trajectory and the ground truth.

src/metrics/nll_loss.py:46

↓ 1 callersClassMlp

MLP as used in Vision Transformer, MLP-Mixer and related networks

src/models/pluto/layers/transformer.py:10

↓ 1 callersClassMlp

src/models/pluto/layers/embedding.py:121

↓ 1 callersClassNATBlock

src/models/pluto/layers/embedding.py:198

↓ 1 callersClassNATLayer

src/models/pluto/layers/embedding.py:147

↓ 1 callersClassNATSequenceEncoder

src/models/pluto/layers/embedding.py:8

↓ 1 callersClassPlanningDecoder

src/models/pluto/modules/planning_decoder.py:89

↓ 1 callersClassRouteManager

src/scenario_manager/route_manager.py:30

↓ 1 callersClassStateAttentionEncoder

src/models/pluto/modules/agent_encoder.py:97

↓ 1 callersClassTrainingEngine

Lightning training engine dataclass wrapping the lightning trainer, model and datamodule.

src/custom_training/custom_training_builder.py:77

↓ 1 callersClassWorldFromPrediction

src/post_processing/observation/world_from_prediction.py:22

↓ 1 callersClassmulADE

Multiple Scope Average Displacement Error mulADE: The average L2 distance between the best forecasted trajectory and the ground truth.

src/metrics/mul_ade.py:13

ClassBBCoordsIndex

src/post_processing/common/enum.py:120

ClassCollisionType

Enum for the types of collisions of interest.

src/post_processing/common/enum.py:147

ClassContrastiveScenarioGenerator

src/data_augmentation/contrastive_scenario_generator.py:29

ClassCrossAttentionLayer

src/models/pluto/layers/transformer.py:97

ClassDynamicStateIndex

src/post_processing/common/enum.py:115

ClassEgoAreaIndex

src/post_processing/common/enum.py:128

ClassLateralStateIndex

Index mapping for the lateral dynamics state vector.

src/post_processing/forward_simulation/batch_lqr.py:26

ClassLightningTrainer

src/models/pluto/scope_trainer.py:30

ClassLightningTrainer

src/models/pluto/pluto_trainer.py:28

ClassMinNormSolver

src/utils/min_norm_solvers.py:5

ClassMultiMetricIndex

src/post_processing/common/enum.py:134

ClassOccupancyType

src/scenario_manager/occupancy_map.py:15

ClassPlanningModel

src/models/pluto/pluto_model.py:27

ClassPlanningModel

src/models/pluto/scope_model.py:28

ClassPlutoPlanner

src/planners/pluto_planner.py:39

ClassPolylineElements

Enum for PolylineElements.

src/feature_builders/common.py:26

ClassScopeContrastiveScenarioGenerator

src/data_augmentation/scope_scenario_generator.py:29

ClassScopeFeatureBuilder

src/feature_builders/scope_feature_builder.py:38

ClassScopePlanner

src/planners/scope_planner.py:39

ClassStateIndex

src/post_processing/common/enum.py:11

ClassTransformerDecoderLayer

src/models/pluto/layers/transformer.py:169

ClassWeightedMetricIndex

src/post_processing/common/enum.py:140