MCPcopy Create free account

hub / github.com/Marine-RL/MarineGym / functions

Functions434 in github.com/Marine-RL/MarineGym

↓ 1 callersMethodcalculate_acc
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:229
↓ 1 callersMethodcalculate_acc
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:285
↓ 1 callersMethodcalculate_added_mass
(self, body_acc)
marinegym/robots/drone/tiltRotor.py:390
↓ 1 callersMethodcalculate_added_mass
(self, body_acc)
marinegym/robots/drone/underwaterVehicle.py:250
↓ 1 callersMethodcalculate_added_mass
(self, body_acc)
marinegym/robots/drone/underwaterVehicleFin.py:306
↓ 1 callersMethodcalculate_buoyancy
(self, rpy)
marinegym/robots/drone/tiltRotor.py:406
↓ 1 callersMethodcalculate_buoyancy
(self, rpy)
marinegym/robots/drone/underwaterVehicle.py:266
↓ 1 callersMethodcalculate_buoyancy
(self, rpy)
marinegym/robots/drone/underwaterVehicleFin.py:322
↓ 1 callersMethodcalculate_corilis
(self, body_vels)
marinegym/robots/drone/tiltRotor.py:396
↓ 1 callersMethodcalculate_corilis
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:256
↓ 1 callersMethodcalculate_corilis
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:312
↓ 1 callersMethodcalculate_damping
(self, body_vels)
marinegym/robots/drone/tiltRotor.py:377
↓ 1 callersMethodcalculate_damping
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:238
↓ 1 callersMethodcalculate_damping
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:294
↓ 1 callersMethodclose
(self)
marinegym/envs/isaac_env.py:232
↓ 1 callersFunctiondefine_prim
Create a USD Prim at the given prim_path of type prim_type unless one already exists Args: prim_path (str): path of the prim in the stage
marinegym/envs/utils/prims.py:75
↓ 1 callersFunctiondict_flatten
Flatten a dict recursively. Examples: >>> a = { "a": 1, "b":{ "c": 3,
marinegym/utils/wandb.py:32
↓ 1 callersFunctiondistance
(p0, p1)
marinegym/utils/poisson_disk.py:52
↓ 1 callersMethodentropy
(self)
marinegym/learning/modules/distributions.py:337
↓ 1 callersFunctionevaluate
( seed: int=0, exploration_type: ExplorationType=ExplorationType.MODE )
scripts/train.py:110
↓ 1 callersFunctionevaluate_model
(env, policy, num_episodes, cfg)
scripts/evaluate.py:55
↓ 1 callersFunctionfits
(p)
marinegym/utils/poisson_disk.py:56
↓ 1 callersMethodget_applied_actions
Gets current applied actions in an ArticulationActions object. Args: clone (bool, optional): True to return clones of the interna
marinegym/views/__init__.py:212
↓ 1 callersFunctionget_first_matching_parent_prim
Recursively get the first USD Prim at the parent path string that passes the predicate function Args: prim_path (str): path of the prim i
marinegym/envs/utils/prims.py:146
↓ 1 callersMethodget_gains
Gets stiffness and damping of articulations in the view. Args: indices (Optional[Union[np.ndarray, List, torch.Tensor]],
marinegym/views/__init__.py:126
↓ 1 callersFunctionget_output_shape
(net, input_size)
marinegym/learning/modules/networks.py:314
↓ 1 callersFunctionget_prim_children
Return the call of the USD Prim's GetChildren member function Args: prim (Usd.Prim): The parent USD Prim Returns: typing.Lis
marinegym/envs/utils/prims.py:240
↓ 1 callersFunctionget_prim_type_name
Get the TypeName of the USD Prim at the path if it is valid Args: prim_path (str): path of the prim in the stage Raises: Exc
marinegym/envs/utils/prims.py:93
↓ 1 callersMethodget_velocities
( self, env_indices: Optional[torch.Tensor] = None, clone: bool = True )
marinegym/views/__init__.py:483
↓ 1 callersMethodget_video_array
(self, axes: str = "t c h w")
marinegym/utils/torchrl/env.py:101
↓ 1 callersMethodget_world_poses
( self, env_indices: Optional[torch.Tensor] = None, clone: bool = True )
marinegym/views/__init__.py:461
↓ 1 callersMethodget_world_poses
(self, env_poses: Tuple[torch.Tensor, torch.Tensor])
marinegym/envs/isaac_env.py:347
↓ 1 callersFunctioninit
(module: nn.Module, weight_init, bias_init, gain=1)
marinegym/learning/modules/distributions.py:39
↓ 1 callersFunctioninit_wandb
Initialize WandB. If only `run_id` is given, resume from the run specified by `run_id`. If only `run_path` is given, start a new run from tha
marinegym/utils/wandb.py:57
↓ 1 callersFunctionis_prim_ancestral
Check if any of the prims ancestors were brought in as a reference Args: prim_path (str): The path to the USD prim. Returns:
marinegym/envs/utils/prims.py:284
↓ 1 callersFunctionis_prim_hidden_in_stage
Checks if the prim is hidden in the USd stage or not. Args: prim_path (str): The path to the USD prim. Note: This is not rel
marinegym/envs/utils/prims.py:323
↓ 1 callersFunctionis_prim_no_delete
Checks whether a prim can be deleted or not from USD stage. Args: prim_path (str): The path to the USD prim. Returns: True i
marinegym/envs/utils/prims.py:311
↓ 1 callersFunctionlemniscate
(t, c)
marinegym/envs/utils/__init__.py:165
↓ 1 callersFunctionload_checkpoint
(checkpoint_path, env_config, algo_config)
scripts/evaluate.py:36
↓ 1 callersFunctionmain
(cfg)
scripts/train.py:36
↓ 1 callersFunctionmain
(cfg)
scripts/evaluate.py:83
↓ 1 callersFunctionmake_batch
(tensordict: TensorDict, num_minibatches: int)
marinegym/learning/ppo/ppo.py:253
↓ 1 callersMethodnormalize
(self, input_vector: torch.Tensor)
marinegym/learning/utils/valuenorm.py:34
↓ 1 callersFunctionorientation_from_view
(camera, target)
marinegym/sensors/camera.py:216
↓ 1 callersMethodreset_parameters
(self)
marinegym/learning/utils/valuenorm.py:67
↓ 1 callersFunctionsample
(p)
marinegym/utils/poisson_disk.py:69
↓ 1 callersFunctionsave_stage
Save usd file to path, it will be overwritten with the current stage Args: usd_path (str): File path to save the current stage to
marinegym/envs/utils/stage.py:237
↓ 1 callersFunctionscale_time
(t, a: float=1.0)
marinegym/envs/utils/__init__.py:199
↓ 1 callersFunctionseparation
(p0, p1, p1_d)
marinegym/robots/drone/tiltRotor.py:605
↓ 1 callersFunctionset_articulation_properties
Set PhysX parameters for an articulation prim. Args: prim_path (str): The prim path to the articulation root. articulation_enable
marinegym/utils/kit.py:229
↓ 1 callersFunctionset_collision_properties
Set PhysX properties of collider prim. Args: prim_path (str): The prim path of parent. collision_enabled (Optional[bool], optiona
marinegym/utils/kit.py:403
↓ 1 callersMethodset_fin_poses
(self, fin_pos)
marinegym/robots/drone/underwaterVehicleFin.py:254
↓ 1 callersMethodset_inertias
( self, values: torch.Tensor, env_indices: Optional[torch.Tensor]=None )
marinegym/views/__init__.py:591
↓ 1 callersFunctionset_rigid_body_properties
Set PhysX parameters for a rigid body prim. Args: prim_path (str): The prim path to the rigid body. rigid_body_enabled (Optional[
marinegym/utils/kit.py:294
↓ 1 callersMethodset_tiltrotor_poses
(self, arm_pos)
marinegym/robots/drone/tiltRotor.py:474
↓ 1 callersMethodset_velocities
( self, velocities: torch.Tensor, env_indices: Optional[torch.Tensor] = None )
marinegym/views/__init__.py:489
↓ 1 callersFunctiontake_first_episode
(tensor: torch.Tensor)
scripts/train.py:137
↓ 1 callersMethodtrain_op
(self, tensordict: TensorDict)
marinegym/learning/ppo/ppo.py:178
↓ 1 callersFunctiontraverse_stage
Traverse through prims in the opened USd stage. Returns: typing.Iterable: Generator which yields prims from the stage in depth-first-trav
marinegym/envs/utils/stage.py:301
Method__call__
(self, env, *args)
marinegym/utils/torchrl/env.py:93
Method__call__
(self, tensordict: TensorDict)
marinegym/learning/ppo/ppo.py:172
Method__init__
(self, name: str, cfg: RobotCfg = None, is_articulation=True)
marinegym/robots/robot.py:61
Method__init__
( self, name: str = None, cfg: TiltRotorCfg=None, is_articulation: bool = T
marinegym/robots/drone/tiltRotor.py:38
Method__init__
( self, name: str = None, cfg: UnderwaterVehicleCfg=None, is_articulation:
marinegym/robots/drone/underwaterVehicle.py:38
Method__init__
( self, name: str = None, cfg: UnderwaterVehicleFinCfg=None, is_articulatio
marinegym/robots/drone/underwaterVehicleFin.py:39
Method__init__
( self, g: float, uav_params, )
marinegym/controllers/lee_position_controller.py:80
Method__init__
(self, g, uav_params)
marinegym/controllers/lee_position_controller.py:216
Method__init__
( self, action_key: Tuple[str] = ("agents", "action"), nbins: Union[int, Sequence[int]
marinegym/utils/torchrl/transforms.py:68
Method__init__
( self, action_key: Tuple[str] = ("agents", "action"), nbins: Union[int, Sequence[int]
marinegym/utils/torchrl/transforms.py:114
Method__init__
( self, in_keys: Sequence[str], min_range: float, max_range: float, in
marinegym/utils/torchrl/transforms.py:158
Method__init__
( self, controller, action_key: str = ("agents", "action"), )
marinegym/utils/torchrl/transforms.py:212
Method__init__
( self, controller, action_key: str = ("agents", "action"), )
marinegym/utils/torchrl/transforms.py:244
Method__init__
(self, interval: int=2)
marinegym/utils/torchrl/env.py:87
Method__init__
(self, in_keys: Sequence[str] = None)
marinegym/utils/torchrl/env.py:106
Method__init__
( self, prim_paths_expr: str, name: str = "articulation_prim_view", positions:
marinegym/views/__init__.py:54
Method__init__
(self)
marinegym/envs/isaac_env.py:49
Method__init__
(self, env: IsaacEnv)
marinegym/envs/isaac_env.py:413
Method__init__
(self, max_t)
marinegym/envs/utils/__init__.py:204
Method__init__
(self, cfg, headless)
marinegym/envs/single/track.py:20
Method__init__
(self, cfg, headless)
marinegym/envs/single/hover.py:18
Method__init__
( self, cfg: Union[PinholeCameraCfg, FisheyeCameraCfg] = None )
marinegym/sensors/camera.py:45
Method__init__
(self, rotor_config, dt: float)
marinegym/actuators/t200.py:6
Method__init__
(self, rotor_config, dt: float)
marinegym/actuators/rotor_group.py:29
Method__init__
(self, rotor_config, dt: float)
marinegym/actuators/m600.py:6
Method__init__
(self, rotor_config, dt: float)
marinegym/actuators/fin200.py:8
Method__init__
( self, input_size, sigma: float = 0.01, fourier_dim=256, embed_dim=72
marinegym/learning/modules/networks.py:97
Method__init__
( self, input_spec: CompositeSpec, embed_dim: int = 72, layer_norm=True,
marinegym/learning/modules/networks.py:126
Method__init__
( self, input_spec: CompositeSpec, *, embed_dim: int = 72, embed_type:
marinegym/learning/modules/networks.py:175
Method__init__
( self, input_spec: CompositeSpec, *, embed_dim: int = 72, embed_type:
marinegym/learning/modules/networks.py:214
Method__init__
( self, input_spec: CompositeSpec, *, query_index=0, embed_dim: int =
marinegym/learning/modules/networks.py:250
Method__init__
( self, cfg, vision_obs_names: List[str] = [], vision_encoder: nn.Module = Non
marinegym/learning/modules/networks.py:320
Method__init__
( self, input_size: torch.Size, )
marinegym/learning/modules/networks.py:370
Method__init__
( self, input_size: int, hidden_size: int )
marinegym/learning/modules/rnn.py:33
Method__init__
(self, num_inputs, num_outputs, use_orthogonal=False, gain=0.01)
marinegym/learning/modules/distributions.py:67
Method__init__
(self, loc, scale, validate_args=None)
marinegym/learning/modules/distributions.py:142
Method__init__
( self, alpha: torch.Tensor, beta: torch.Tensor, min: Union[float, torch.Tenso
marinegym/learning/modules/distributions.py:149
Method__init__
( self, input_dim: int, output_dim: int, state_dependent_std: bool = False,
marinegym/learning/modules/distributions.py:177
Method__init__
( self, input_dim: int, output_dim: int, scale_mapping: str = "softplus",
marinegym/learning/modules/distributions.py:218
Method__init__
( self, input_dim: int, output_dim: int, scale_mapping: str = "softplus",
marinegym/learning/modules/distributions.py:258
Method__init__
( self, logits: List[torch.Tensor] = None, probs: List[torch.Tensor] = None, )
marinegym/learning/modules/distributions.py:287
← previousnext →101–200 of 434, ranked by callers