MCPcopy Create free account

hub / github.com/FrankZxShen/MCoCoNav / functions

Functions1,026 in github.com/FrankZxShen/MCoCoNav

↓ 25 callersMethodexists
(self)
multi-robot-setting/habitat/tasks/rearrange/utils.py:249
↓ 24 callersMethodget
(self, uuid: str)
multi-robot-setting/habitat/core/simulator.py:207
↓ 20 callersMethodget
(self)
main.py:67
↓ 18 callersMethodget_agent_state
r""".. :param agent_id: id of agent. :return: state of agent corresponding to :p:`agent_id`.
multi-robot-setting/habitat/core/simulator.py:290
↓ 15 callersMethod__init__
(self, *args, sim, config, task, **kwargs)
multi-robot-setting/habitat/tasks/rearrange/rearrange_sensors.py:512
↓ 15 callersMethodupdate
(self)
multi-robot-setting/habitat/tasks/rearrange/marker_info.py:38
↓ 14 callersMethodget_observations
r"""Collects data from all sensors and returns it packaged inside :ref:`Observations`.
multi-robot-setting/habitat/core/simulator.py:210
↓ 11 callersMethod__init__
( self, sim: Simulator, config: Config, *args: Any, **kwargs: Any )
multi-robot-setting/habitat/tasks/nav/nav.py:629
↓ 11 callersFunctiond240
(x)
main.py:366
↓ 11 callersMethodextend_action_space
r"""Extends the action space to accommodate a new action with the name :p:`name` :param name: The name of the new action :ret
multi-robot-setting/habitat/sims/habitat_simulator/actions.py:45
↓ 11 callersMethodget_metric
r""".. :return: the current metric for :ref:`Measure`.
multi-robot-setting/habitat/core/embodied_task.py:117
↓ 10 callersMethodget
(self)
multi-robot-setting/habitat/sims/habitat_simulator/actions.py:94
↓ 9 callersMethodhas_action
r"""Checks to see if action :p:`name` is already register :param name: The name to check :return: Whether or not :p:`name` already ex
multi-robot-setting/habitat/sims/habitat_simulator/actions.py:67
↓ 9 callersMethodseed
(self, seed: int)
multi-robot-setting/habitat/core/env.py:351
↓ 8 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
RedNet/RedNet_model.py:81
↓ 8 callersFunctioncoll_name_matches
(coll, name)
multi-robot-setting/habitat/tasks/rearrange/utils.py:40
↓ 8 callersMethodget_spl
This function computes evaluation metrics for the Object Goal task Returns: spl (float): Success weighted by Path Length
agents/vlm_agents.py:1110
↓ 8 callersMethodget_targets
Get a mapping of object ids to goal positions for rearrange targets. :return: ([idx: int], [goal_pos: list]) The index of the target object
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:571
↓ 8 callersMethodload
(self, load_depth=0)
multi-robot-setting/habitat/tasks/rearrange/utils.py:252
↓ 7 callersMethod_get_impl
(cls, _type: str, name: str)
multi-robot-setting/habitat/core/registry.py:198
↓ 7 callersMethod_register_impl
( cls, _type: str, to_register: Optional[Any], name: Optional[str], as
multi-robot-setting/habitat/core/registry.py:43
↓ 7 callersMethodcheck_measure_dependencies
r"""Checks if dependencies measures are enabled and calculatethat the measure :param measure_name: a name of the measure for which has depende
multi-robot-setting/habitat/core/embodied_task.py:177
↓ 7 callersMethodclose
(self)
multi-robot-setting/habitat/core/env.py:372
↓ 7 callersMethodget_observations_at
Returns the observation. :param position: list containing 3 entries for :py:`(x, y, z)`. :param rotation: list with 4 entries for :py
multi-robot-setting/habitat/core/simulator.py:298
↓ 7 callersMethodget_scene_pos
Get the positions of all clutter RigidObjects in the scene as a numpy array.
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:594
↓ 7 callersFunctionnormalize
(v)
utils/rotation_utils.py:23
↓ 7 callersMethodstep
r"""Update ``_metric``, this method is called from ``Env`` on each ``step``.
multi-robot-setting/habitat/tasks/nav/nav.py:1136
↓ 7 callersMethodstep
(self, arm_action, grip_action=None, *args, **kwargs)
multi-robot-setting/habitat/tasks/rearrange/actions.py:91
↓ 6 callersMethodclose
(self)
multi-robot-setting/habitat/core/vector_env.py:454
↓ 6 callersMethodget_observation
r""" Returns: current observation for Sensor.
multi-robot-setting/habitat/core/simulator.py:88
↓ 6 callersMethodsave
(self, val)
multi-robot-setting/habitat/tasks/rearrange/utils.py:273
↓ 5 callersMethod__init__
(self, *args: Any, **kwargs: Any)
multi-robot-setting/habitat/core/simulator.py:68
↓ 5 callersMethod_make_agant_layer
(self, inplanes, planes)
RedNet/RedNet_model.py:126
↓ 5 callersMethod_make_transpose
(self, block, planes, blocks, stride=1)
RedNet/RedNet_model.py:99
↓ 5 callersMethodinternal_step
Step the world and update the robot. :param dt: Timestep by which to advance the world. Multiple physics substeps can be excecuted within a s
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:555
↓ 5 callersFunctionquaternion_from_coeff
r"""Creates a quaternions from coeffs in [x, y, z, w] format
multi-robot-setting/habitat/utils/geometry_utils.py:54
↓ 5 callersFunctionquaternion_rotate_vector
r"""Rotates a vector by a quaternion Args: quaternion: The quaternion to rotate by v: The vector to rotate Returns: np
multi-robot-setting/habitat/utils/geometry_utils.py:62
↓ 5 callersMethodreset
r"""Resets the environments and returns the initial observations. :return: initial observations from the environment.
multi-robot-setting/habitat/core/env.py:228
↓ 5 callersFunctiontry_cv2_import
r"""The PyRobot python3 version which is a dependency of Habitat-PyRobot integration relies on ROS running in python2.7. In order to import cv2 in
multi-robot-setting/habitat/core/utils.py:70
↓ 4 callersMethod__init__
(self, num_classes=40, pretrained=False)
RedNet/RedNet_model.py:13
↓ 4 callersMethod_get_agent_config
(self, agent_id: Optional[int] = None)
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:535
↓ 4 callersMethod_try_acquire_context
(self)
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:98
↓ 4 callersFunctioncartesian_to_polar
(x, y)
multi-robot-setting/habitat/tasks/utils.py:39
↓ 4 callersMethodcreate_chat_completion
This function sends a request to the chat API to generate a response based on the given messages. Args: model (str): The
src/vlm.py:25
↓ 4 callersMethodcreate_hold_constraint
Create a new rigid point-to-point (ball joint) constraint between the robot and an object. :param pivot_in_link: The origin of the constraint
multi-robot-setting/habitat/tasks/rearrange/rearrange_grasp_manager.py:186
↓ 4 callersMethoddesnap
Removes any hold constraints currently active. :param force: If True, reset the collision group of the now released object immediately instea
multi-robot-setting/habitat/tasks/rearrange/rearrange_grasp_manager.py:110
↓ 4 callersMethodencode_image
Encodes an image file into a base64 string. Args: image_path (str): The path to the image file. This function op
src/vlm.py:77
↓ 4 callersMethodgeodesic_distance
r"""Calculates geodesic distance between two points. :param position_a: coordinates of first point. :param position_b: coordinates of
multi-robot-setting/habitat/core/simulator.py:268
↓ 4 callersFunctionget_r_matrix
(ax_, angle)
utils/rotation_utils.py:27
↓ 4 callersMethodget_target_objs_start
Get the initial positions of all objects targeted for rearrangement as a numpy array.
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:590
↓ 4 callersMethodlook_at
Point the debug camera at a target.
multi-robot-setting/habitat/sims/habitat_simulator/debug_visualizer.py:40
↓ 4 callersMethodreset
(self, *args, **kwargs)
multi-robot-setting/habitat/tasks/rearrange/actions.py:77
↓ 4 callersMethodreset
(self)
multi-robot-setting/habitat/datasets/rearrange/samplers.py:26
↓ 4 callersMethodsample
(self)
multi-robot-setting/habitat/core/spaces.py:107
↓ 4 callersMethodsample
(self)
multi-robot-setting/habitat/datasets/rearrange/samplers.py:30
↓ 4 callersMethodset_agent_state
r"""Sets agent state similar to initialize_agent, but without agents creation. On failure to place the agent in the proper position, it is
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:545
↓ 4 callersMethodset_multi_goal
(self, goal_map)
utils/fmm_planner.py:81
↓ 4 callersMethodsimple_image_chat
Facilitates a simple chat interaction involving an image. COT1 Args: use_stream (bool): Specifies whether to use
src/vlm.py:101
↓ 3 callersFunctionPerception_weight_decision
(VLM_Rel: list, VLM_Pred: str)
src/SystemPrompt.py:435
↓ 3 callersMethod__init__
(self, sim, config, *args: Any, **kwargs: Any)
multi-robot-setting/habitat/tasks/eqa/eqa.py:98
↓ 3 callersMethod__init__
(self, config: Config)
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:254
↓ 3 callersMethod_compute_pointgoal
( self, source_position, source_rotation, goal_position )
multi-robot-setting/habitat/tasks/nav/nav.py:233
↓ 3 callersMethod_draw_point
(self, position, point_type)
multi-robot-setting/habitat/tasks/nav/nav.py:797
↓ 3 callersMethod_init_entities
( self, entity_names, register_func, entities_config=None )
multi-robot-setting/habitat/core/embodied_task.py:258
↓ 3 callersMethod_is_on_same_floor
( self, height, ref_floor_height=None, ceiling_height=2.0 )
multi-robot-setting/habitat/tasks/nav/nav.py:911
↓ 3 callersMethodbuild_content_scenes_filter
r"""Returns a filter function that takes an episode and returns True if that episode is valid under the CONTENT_SCENES feild of the provided c
multi-robot-setting/habitat/core/dataset.py:130
↓ 3 callersFunctioncheck_sim_obs
(obs: Optional[np.ndarray], sensor: Sensor)
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:237
↓ 3 callersFunctionfind_big_connect
(image)
main.py:292
↓ 3 callersMethodget_metrics
(self)
multi-robot-setting/habitat/core/env.py:211
↓ 3 callersMethodget_observation
Render a debug observation of the current state and cache it. Optionally configure the camera transform. Optionally provide a
multi-robot-setting/habitat/sims/habitat_simulator/debug_visualizer.py:68
↓ 3 callersMethodget_robot_observations
(self)
multi-robot-setting/habitat/sims/pyrobot/pyrobot.py:202
↓ 3 callersMethodis_violating_hold_constraint
Returns true if the object is too far away from the gripper, meaning the agent violated the hold constraint.
multi-robot-setting/habitat/tasks/rearrange/rearrange_grasp_manager.py:64
↓ 3 callersMethodisland_radius
(self, position: Sequence[float])
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:620
↓ 3 callersFunctionoverwrite_config
r"""Takes Habitat Lab config and Habitat-Sim config structures. Overwrites Habitat-Sim config with Habitat Lab values, where a field name is prese
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:46
↓ 3 callersFunctionparse_receptacles_from_user_config
Parse receptacle metadata from the provided user subconfig object. :param user_subconfig: The Configuration object containing metadata parse
multi-robot-setting/habitat/datasets/rearrange/receptacle.py:270
↓ 3 callersFunctionrearrange_collision
Defines what counts as a collision for the Rearrange environment execution
multi-robot-setting/habitat/tasks/rearrange/utils.py:87
↓ 3 callersMethodreset
r"""Reset all the vectorized environments :return: list of outputs from the reset method of envs.
multi-robot-setting/habitat/core/vector_env.py:375
↓ 3 callersMethodsemantic_annotations
r""" Returns: SemanticScene which is a three level hierarchy of semantic annotations for the current scene. Specifical
multi-robot-setting/habitat/sims/habitat_simulator/habitat_simulator.py:503
↓ 3 callersMethodset_state
Sets the simulation state from a cached state info dict. See capture_state(). :param set_hold: If true this will set the snapped o
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:416
↓ 3 callersMethodsnap_to_obj
Attempt to grasp an object, snapping/constraining it to the robot's end effector with 3 ball-joint constraints forming a fixed frame. :param
multi-robot-setting/habitat/tasks/rearrange/rearrange_grasp_manager.py:213
↓ 3 callersMethodstep
r"""Step method is called from ``Env`` on each ``step``. Can call simulator or task method, change task's state. :param kwargs: optio
multi-robot-setting/habitat/core/embodied_task.py:40
↓ 3 callersMethodstep
r"""Perform an action in the environment and return observations. :param action: action (belonging to :ref:`action_space`) to be
multi-robot-setting/habitat/core/env.py:294
↓ 3 callersMethodvisualize_position
Adds the sphere object to the specified position for visualization purpose. :param position: global position of the visual sphere :pa
multi-robot-setting/habitat/tasks/rearrange/rearrange_sim.py:520
↓ 2 callersMethodCoT2
Facilitates a simple chat interaction involving an image. Args: use_stream (bool): Specifies whether to use streaming fo
src/vlm.py:146
↓ 2 callersFunctionDecision_Generation_Vis
(args, agents_seg_list, agent_j, episode_n, l_step, pose_pred, full_map_pred, goal_name,
main.py:403
↓ 2 callersFunctionFrontiers
(full_map_pred)
main.py:172
↓ 2 callersMethod__getstate__
(self)
multi-robot-setting/habitat/core/dataset.py:85
↓ 2 callersMethod__init__
(self, *args, config, sim: RearrangeSim, **kwargs)
multi-robot-setting/habitat/tasks/rearrange/actions.py:57
↓ 2 callersMethod__init__
(self, config: Config)
multi-robot-setting/habitat/sims/pyrobot/pyrobot.py:170
↓ 2 callersMethod_build
(self)
multi-robot-setting/habitat/datasets/utils.py:70
↓ 2 callersMethod_check_episode_is_active
( self, *args: Any, action: Union[int, Dict[str, Any]], episode: Episode,
multi-robot-setting/habitat/core/embodied_task.py:385
↓ 2 callersMethod_euclidean_distance
(self, position_a, position_b)
multi-robot-setting/habitat/tasks/nav/nav.py:660
↓ 2 callersMethod_get_measure_index
(self, measure_name)
multi-robot-setting/habitat/core/embodied_task.py:174
↓ 2 callersMethod_get_scenes_from_folder
( content_scenes_path: str, dataset_dir: str )
multi-robot-setting/habitat/datasets/pointnav/pointnav_dataset.py:69
↓ 2 callersMethod_group_scenes
r"""Internal method that groups episodes by scene Groups will be ordered by the order the first episode of a given scene is in the lis
multi-robot-setting/habitat/core/dataset.py:486
↓ 2 callersMethod_peek_object
Compute a camera placement to view an ArticulatedObject and show/save an observation. Return the filepath. If peek_all_axis,
multi-robot-setting/habitat/sims/habitat_simulator/debug_visualizer.py:154
↓ 2 callersMethod_preprocess_depth
(self, depth, min_d, max_d)
agents/vlm_agents.py:783
↓ 2 callersMethod_preprocess_depth
(self, depth, min_d, max_d)
agents/vlm_agents_gt.py:806
↓ 2 callersMethod_preprocess_semantic
(self, semantic)
envs/habitat/multi_agent_sem_env.py:89
↓ 2 callersMethod_quat_to_xy_heading
(self, quat)
multi-robot-setting/habitat/tasks/nav/nav.py:426
next →1–100 of 1,026, ranked by callers