Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Marine-RL/MarineGym
/ functions
Functions
434 in github.com/Marine-RL/MarineGym
⨍
Functions
434
◇
Types & classes
73
↓ 1 callers
Method
calculate_acc
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:229
↓ 1 callers
Method
calculate_acc
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:285
↓ 1 callers
Method
calculate_added_mass
(self, body_acc)
marinegym/robots/drone/tiltRotor.py:390
↓ 1 callers
Method
calculate_added_mass
(self, body_acc)
marinegym/robots/drone/underwaterVehicle.py:250
↓ 1 callers
Method
calculate_added_mass
(self, body_acc)
marinegym/robots/drone/underwaterVehicleFin.py:306
↓ 1 callers
Method
calculate_buoyancy
(self, rpy)
marinegym/robots/drone/tiltRotor.py:406
↓ 1 callers
Method
calculate_buoyancy
(self, rpy)
marinegym/robots/drone/underwaterVehicle.py:266
↓ 1 callers
Method
calculate_buoyancy
(self, rpy)
marinegym/robots/drone/underwaterVehicleFin.py:322
↓ 1 callers
Method
calculate_corilis
(self, body_vels)
marinegym/robots/drone/tiltRotor.py:396
↓ 1 callers
Method
calculate_corilis
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:256
↓ 1 callers
Method
calculate_corilis
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:312
↓ 1 callers
Method
calculate_damping
(self, body_vels)
marinegym/robots/drone/tiltRotor.py:377
↓ 1 callers
Method
calculate_damping
(self, body_vels)
marinegym/robots/drone/underwaterVehicle.py:238
↓ 1 callers
Method
calculate_damping
(self, body_vels)
marinegym/robots/drone/underwaterVehicleFin.py:294
↓ 1 callers
Method
close
(self)
marinegym/envs/isaac_env.py:232
↓ 1 callers
Function
define_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 callers
Function
dict_flatten
Flatten a dict recursively. Examples: >>> a = { "a": 1, "b":{ "c": 3,
marinegym/utils/wandb.py:32
↓ 1 callers
Function
distance
(p0, p1)
marinegym/utils/poisson_disk.py:52
↓ 1 callers
Method
entropy
(self)
marinegym/learning/modules/distributions.py:337
↓ 1 callers
Function
evaluate
( seed: int=0, exploration_type: ExplorationType=ExplorationType.MODE )
scripts/train.py:110
↓ 1 callers
Function
evaluate_model
(env, policy, num_episodes, cfg)
scripts/evaluate.py:55
↓ 1 callers
Function
fits
(p)
marinegym/utils/poisson_disk.py:56
↓ 1 callers
Method
get_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 callers
Function
get_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 callers
Method
get_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 callers
Function
get_output_shape
(net, input_size)
marinegym/learning/modules/networks.py:314
↓ 1 callers
Function
get_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 callers
Function
get_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 callers
Method
get_velocities
( self, env_indices: Optional[torch.Tensor] = None, clone: bool = True )
marinegym/views/__init__.py:483
↓ 1 callers
Method
get_video_array
(self, axes: str = "t c h w")
marinegym/utils/torchrl/env.py:101
↓ 1 callers
Method
get_world_poses
( self, env_indices: Optional[torch.Tensor] = None, clone: bool = True )
marinegym/views/__init__.py:461
↓ 1 callers
Method
get_world_poses
(self, env_poses: Tuple[torch.Tensor, torch.Tensor])
marinegym/envs/isaac_env.py:347
↓ 1 callers
Function
init
(module: nn.Module, weight_init, bias_init, gain=1)
marinegym/learning/modules/distributions.py:39
↓ 1 callers
Function
init_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 callers
Function
is_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 callers
Function
is_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 callers
Function
is_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 callers
Function
lemniscate
(t, c)
marinegym/envs/utils/__init__.py:165
↓ 1 callers
Function
load_checkpoint
(checkpoint_path, env_config, algo_config)
scripts/evaluate.py:36
↓ 1 callers
Function
main
(cfg)
scripts/train.py:36
↓ 1 callers
Function
main
(cfg)
scripts/evaluate.py:83
↓ 1 callers
Function
make_batch
(tensordict: TensorDict, num_minibatches: int)
marinegym/learning/ppo/ppo.py:253
↓ 1 callers
Method
normalize
(self, input_vector: torch.Tensor)
marinegym/learning/utils/valuenorm.py:34
↓ 1 callers
Function
orientation_from_view
(camera, target)
marinegym/sensors/camera.py:216
↓ 1 callers
Method
reset_parameters
(self)
marinegym/learning/utils/valuenorm.py:67
↓ 1 callers
Function
sample
(p)
marinegym/utils/poisson_disk.py:69
↓ 1 callers
Function
save_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 callers
Function
scale_time
(t, a: float=1.0)
marinegym/envs/utils/__init__.py:199
↓ 1 callers
Function
separation
(p0, p1, p1_d)
marinegym/robots/drone/tiltRotor.py:605
↓ 1 callers
Function
set_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 callers
Function
set_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 callers
Method
set_fin_poses
(self, fin_pos)
marinegym/robots/drone/underwaterVehicleFin.py:254
↓ 1 callers
Method
set_inertias
( self, values: torch.Tensor, env_indices: Optional[torch.Tensor]=None )
marinegym/views/__init__.py:591
↓ 1 callers
Function
set_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 callers
Method
set_tiltrotor_poses
(self, arm_pos)
marinegym/robots/drone/tiltRotor.py:474
↓ 1 callers
Method
set_velocities
( self, velocities: torch.Tensor, env_indices: Optional[torch.Tensor] = None )
marinegym/views/__init__.py:489
↓ 1 callers
Function
take_first_episode
(tensor: torch.Tensor)
scripts/train.py:137
↓ 1 callers
Method
train_op
(self, tensordict: TensorDict)
marinegym/learning/ppo/ppo.py:178
↓ 1 callers
Function
traverse_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
← previous
next →
101–200 of 434, ranked by callers