Functions — github.com/Rex-sys-hk/PlanScope

hub / github.com/Rex-sys-hk/PlanScope / functions

Functions459 in github.com/Rex-sys-hk/PlanScope

⨍ Functions 459◇ Types & classes 70

Methodneg_interacting_agent_dropout

(self, data)

src/data_augmentation/contrastive_scenario_generator.py:213

Methodneg_interacting_agent_dropout

(self, data)

src/data_augmentation/scope_scenario_generator.py:215

Methodneg_leading_agent_insertion

(self, data)

src/data_augmentation/contrastive_scenario_generator.py:223

Methodneg_leading_agent_insertion

(self, data)

src/data_augmentation/scope_scenario_generator.py:225

Methodneg_traffic_light_inversion

(self, data)

src/data_augmentation/contrastive_scenario_generator.py:205

Methodneg_traffic_light_inversion

(self, data)

src/data_augmentation/scope_scenario_generator.py:207

Methodnormalize

( self, data, first_time=False, radius=None, hist_steps=21 )

src/features/scope_feature.py:168

Functionnormalize_angle

(angle)

src/planners/ml_planner_utils.py:17

Methodobservation_type

Inherited, see superclass.

src/planners/pluto_planner.py:149

Methodobservation_type

Inherited, see superclass.

src/planners/scope_planner.py:151

Methodon_fit_start

(self)

src/models/pluto/scope_trainer.py:107

Methodon_fit_start

(self)

src/models/pluto/pluto_trainer.py:74

Functionplot_actor_bounding_box

Plot an actor bounding box centered on the actor's current location. Args: ax: Axes on which actor bounding box should be plotted.

src/utils/vis.py:27

Functionplot_cov_ellipse

Plots an `nstd` sigma error ellipse based on the specified covariance matrix (`cov`). Additional keyword arguments are passed on to the e

src/utils/vis.py:319

Functionplot_crosswalk

(ax, edge1, edge2)

src/utils/vis.py:258

Functionplot_direction

(ax, anchors, dir_vecs, zorder=1)

src/utils/vis.py:215

Functionplot_lane_area

(ax, left_bound, right_bound, fc="silver", alpha=1.0, ec=None)

src/utils/vis.py:312

Functionplot_polygon

( ax: plt.Axes, x: np.ndarray, y: np.ndarray, color: str, alpha=1.0, zorder=50, la

src/utils/vis.py:118

Functionplot_sdc

( ax, center, heading, width, length, steer=0.0, color="pink", fill=True,

src/utils/vis.py:267

Functionplot_trajectory_with_angle

(ax, traj)

src/utils/vis.py:232

Methodpoints_in_polygons

Determines wether input-points are in polygons of the occupancy map :param points: input-points :return: boolean array of sha

src/scenario_manager/occupancy_map.py:111

Functionprint_dict_tensor

(data, prefix="")

src/utils/utils.py:65

Functionprint_simulation_results

(file=None)

src/utils/utils.py:74

Methodrender_from_scenario

( self, scenario: AbstractScenario, ego_state: EgoState = None, iteration=0,

src/feature_builders/nuplan_scenario_render.py:140

Methodrequired_features

Inherited, see superclass.

src/data_augmentation/contrastive_scenario_generator.py:311

Methodrequired_features

Inherited, see superclass.

src/data_augmentation/scope_scenario_generator.py:313

Methodrequired_targets

Inherited, see superclass.

src/data_augmentation/contrastive_scenario_generator.py:316

Methodrequired_targets

Inherited, see superclass.

src/data_augmentation/scope_scenario_generator.py:318

Functionrotate_round_z_axis

(points: np.ndarray, angle: float)

src/feature_builders/common.py:120

Methodroute_lane_ids

(self)

src/scenario_manager/route_manager.py:50

Functionsafe_index

(ls, value)

src/utils/utils.py:93

Methodserialize

(self)

src/features/scope_feature.py:150

Methodsetup

Set up the dataset for each target set depending on the training stage. This is called by every process in distributed training.

src/custom_training/custom_datamodule.py:174

Methodstate_dict

Returns the state of the scheduler as a :class:`dict`. It contains an entry for every variable in self.__dict__ which is not the opti

src/optim/warmup_cos_lr.py:16

Methodteardown

Clean up after a training stage. This is called by every process in distributed training. :param stage: Stage of training, ca

src/custom_training/custom_datamodule.py:236

Methodtest_dataloader

Create the test dataloader. :raises RuntimeError: if this method is called without calling "setup()" first. :return: The inst

src/custom_training/custom_datamodule.py:285

Methodtest_step

Step called for each batch example during testing. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/scope_trainer.py:439

Methodtest_step

Step called for each batch example during testing. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/pluto_trainer.py:380

Methodto_device

(self, device)

src/utils/collision_checker.py:27

Methodto_device

(self, device: torch.device)

src/features/scope_feature.py:144

Methodto_feature_tensor

(self)

src/features/scope_feature.py:96

Methodto_numpy

(self)

src/features/pluto_feature.py:126

Methodto_numpy

(self)

src/features/scope_feature.py:126

Methodtoken_to_idx

Getter for track tokens in occupancy map :return: dictionary of tokens and indices

src/scenario_manager/occupancy_map.py:73

Methodtokens

Getter for track tokens in occupancy map :return: list of strings

src/scenario_manager/occupancy_map.py:65

Methodtrain_dataloader

Create the training dataloader. :raises RuntimeError: If this method is called without calling "setup()" first. :return: The

src/custom_training/custom_datamodule.py:244

Methodtraining_step

Step called for each batch example during training. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/scope_trainer.py:415

Methodtraining_step

Step called for each batch example during training. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/pluto_trainer.py:356

Methodtransfer_batch_to_device

Transfer a batch to device. :param batch: Batch on origin device. :param device: Desired device. :return: Batch in ne

src/custom_training/custom_datamodule.py:301

Methodunpack

(self)

src/features/pluto_feature.py:157

Methodunpack

(self)

src/features/scope_feature.py:157

Methodupdate

outputs: [A, T, 2] target: [A, T, 2]

src/metrics/prediction_avg_ade.py:28

Methodupdate

outputs: [A, T, 2] target: [A, T, 2]

src/metrics/prediction_avg_fde.py:28

Methodupdate

(self, outputs: Dict[str, torch.Tensor], target: torch.Tensor)

src/metrics/min_fde.py:31

Methodupdate

(self, outputs: Dict[str, torch.Tensor], target: torch.Tensor)

src/metrics/min_ade.py:36

Methodupdate

( self, ego_state: EgoState, detections: DetectionsTracks, traffic_light_data:

src/post_processing/observation/world_from_prediction.py:49

Methodval_dataloader

Create the validation dataloader. :raises RuntimeError: if this method is called without calling "setup()" first. :return: Th

src/custom_training/custom_datamodule.py:270

Methodvalidation_step

Step called for each batch example during validation. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/scope_trainer.py:427

Methodvalidation_step

Step called for each batch example during validation. :param batch: example batch :param batch_idx: batch's index (unused)

src/models/pluto/pluto_trainer.py:368

← previous401–459 of 459, ranked by callers