MCPcopy Create free account

hub / github.com/bagh2178/UniGoal / functions

Functions241 in github.com/bagh2178/UniGoal

↓ 1 callersMethodgoal_decomposition
(self, goalgraph=None)
src/graph/goalgraphdecomposer.py:33
↓ 1 callersFunctiongobs_to_detection_list
Return a DetectionList object from the gobs All object are still in the camera frame.
src/graph/utils/utils.py:199
↓ 1 callersMethodgraph_corr
(self, goal, graph)
src/graph/graph.py:921
↓ 1 callersMethodgraph_corr
(self, nodes_to_check, edges_to_check)
src/graph/scenegraphcorrector.py:51
↓ 1 callersMethodgraph_to_text
(self, graph)
src/graph/goalgraphdecomposer.py:24
↓ 1 callersMethodinit_map_and_pose
(self)
src/map/bev_mapping.py:289
↓ 1 callersMethodinit_map_and_pose_for_env
(self, env_idx=0)
src/map/bev_mapping.py:316
↓ 1 callersFunctioninit_vis_image
(goal_name, args)
src/utils/visualization/visualization.py:33
↓ 1 callersMethodinsert_goal
(self, goal=None)
src/graph/graph.py:655
↓ 1 callersMethodinstance_discriminator
(self, planner_inputs, id_lo_whwh_speci)
src/agent/unigoal/agent.py:195
↓ 1 callersMethodis_edge_reasonable
(self, edge, source_node, target_node)
src/graph/scenegraphcorrector.py:42
↓ 1 callersMethodis_node_reasonable
(self, node, connected_nodes, connected_edges)
src/graph/scenegraphcorrector.py:30
↓ 1 callersFunctionline_list
(text, line_length=22)
src/utils/visualization/visualization.py:100
↓ 1 callersFunctionmain
()
main.py:57
↓ 1 callersMethodmapping3d
(self)
src/graph/graph.py:454
↓ 1 callersFunctionmask_subtract_contained
Compute the containing relationship between all pair of bounding boxes. For each mask, subtract the mask of bounding boxes that are contained
src/graph/utils/iou.py:108
↓ 1 callersFunctionmerge_detections_to_objects
( cfg, detection_list: DetectionList, objects: MapObjectList, agg_sim: torch.Tensor )
src/graph/utils/mapping.py:29
↓ 1 callersFunctionmerge_obj2_into_obj1
Merge the new object to the old object This operation is done in-place
src/graph/utils/utils.py:432
↓ 1 callersMethodmove_local_map
(self, env_idx=0)
src/map/bev_mapping.py:246
↓ 1 callersFunctionmoving_avg
(a, n=2)
src/utils/fmm/fmm_planner.py:53
↓ 1 callersMethodnode_similarity
(self, G1, G2)
src/graph/overlap.py:30
↓ 1 callersMethodoverlap
(self)
src/graph/graph.py:714
↓ 1 callersMethodparse_text_description
(self, description)
src/graph/graphbuilder.py:31
↓ 1 callersFunctionpcd_denoise_dbscan
(pcd: o3d.geometry.PointCloud, eps=0.02, min_points=10)
src/graph/utils/utils.py:364
↓ 1 callersMethodpredict_remaining_node_positions
(self, common_nodes, positions, scene_graph)
src/graph/overlap.py:123
↓ 1 callersMethodpreprocess_depth
(self, depth, min_d, max_d)
src/agent/unigoal/agent.py:647
↓ 1 callersFunctionr_between
(v_from_, v_to_)
src/utils/fmm/rotation_utils.py:40
↓ 1 callersMethodreasonableness_correction
(self)
src/graph/graph.py:740
↓ 1 callersFunctionremove_small_frontiers
(frontier_map, min_size=1)
src/utils/map.py:4
↓ 1 callersFunctionresize_gobs
( gobs, image )
src/graph/utils/utils.py:76
↓ 1 callersMethodrun_on_image
Args: image (np.ndarray): an image of shape (H, W, C) (in BGR order). This is the format used by OpenCV.
src/utils/visualization/semantic_prediction.py:155
↓ 1 callersFunctionsave_video
(image_list, video_path, fps=30, resize_factor=1.0, input_color_space="RGB")
src/utils/visualization/save.py:6
↓ 1 callersMethodsave_visualization
(self, video_path)
src/agent/unigoal/agent.py:841
↓ 1 callersMethodseed
(self, seed)
src/envs/instanceimagegoal_env.py:126
↓ 1 callersMethodsegment2d
(self)
src/graph/graph.py:422
↓ 1 callersMethodset_center
(self, center)
src/graph/graph.py:128
↓ 1 callersMethodset_cfg
(self)
src/graph/graph.py:255
↓ 1 callersMethodset_full_map
(self, full_map)
src/graph/graph.py:302
↓ 1 callersMethodset_full_pose
(self, full_pose)
src/graph/graph.py:305
↓ 1 callersMethodset_goal
(self, goal, auto_improve=False)
src/utils/fmm/fmm_planner_policy.py:62
↓ 1 callersMethodset_goal_cat_id
(self, idx)
src/envs/instanceimagegoal_env.py:129
↓ 1 callersMethodset_navigate_steps
(self, navigate_steps)
src/graph/graph.py:267
↓ 1 callersMethodset_object
(self, object)
src/graph/graph.py:124
↓ 1 callersMethodset_observations
(self, observations)
src/graph/graph.py:276
↓ 1 callersFunctionsetup_cfg
(args)
src/utils/visualization/semantic_prediction.py:87
↓ 1 callersFunctionsplat_feat_nd
Args: init_grid: B X nF X W X H X D X .. feat: B X nF X nPt coords: B X nDims X nPt in [-1, 1] Returns: grid:
src/utils/fmm/depth_utils.py:200
↓ 1 callersMethodtext2value
(self, text)
src/graph/graph.py:933
↓ 1 callersFunctiontransform_camera_view_t
Transforms the point cloud into geocentric frame to account for camera elevation and angle Input: XYZ : ...x3
src/utils/fmm/depth_utils.py:160
↓ 1 callersFunctiontransform_pose_t
Transforms the point cloud into geocentric frame to account for camera position Input: XYZ : ...x3 cu
src/utils/fmm/depth_utils.py:181
↓ 1 callersMethodupdate_after_reset
(self)
src/envs/instanceimagegoal_env.py:50
↓ 1 callersMethodupdate_edge
(self)
src/graph/graph.py:577
↓ 1 callersMethodupdate_group
(self)
src/graph/graph.py:638
↓ 1 callersMethodupdate_node
(self)
src/graph/graph.py:516
↓ 1 callersMethodupdate_scenegraph
(self)
src/graph/graph.py:682
↓ 1 callersMethodvisualize
(self, inputs)
src/agent/unigoal/agent.py:709
Method__add__
(self, other)
src/graph/utils/slam_classes.py:48
Method__call__
(self, prompt)
src/utils/llm.py:12
Method__call__
(self, prompt, image)
src/utils/llm.py:32
Method__call__
Args: image_list (list of np.ndarray): a list of images of shape (H, W, C) (in BGR o
src/utils/visualization/semantic_prediction.py:225
Method__iadd__
(self, other)
src/graph/utils/slam_classes.py:53
Method__init__
(self, base_url, api_key, llm_model)
src/utils/llm.py:7
Method__init__
(self, base_url, api_key, vlm_model)
src/utils/llm.py:27
Method__init__
(self, args)
src/utils/visualization/semantic_prediction.py:24
Method__init__
(self, args)
src/utils/visualization/semantic_prediction.py:62
Method__init__
Args: cfg (CfgNode): instance_mode (ColorMode):
src/utils/visualization/semantic_prediction.py:141
Method__init__
(self, cfg)
src/utils/visualization/semantic_prediction.py:213
Method__init__
(self, traversible, scale=1, step_size=5)
src/utils/fmm/fmm_planner.py:68
Method__init__
(self, traversible, scale=1, step_size=5)
src/utils/fmm/fmm_planner_policy.py:47
Method__init__
(self, args)
src/envs/real_world_env.py:6
Method__init__
(self, args, config_env=None, dataset=None)
src/envs/instanceimagegoal_env.py:17
Method__init__
(self, args)
src/map/bev_mapping.py:17
Method__init__
(self, args)
src/map/bev_mapping.py:186
Method__init__
(self, graph1, graph2, llm=None)
src/graph/overlap.py:11
Method__init__
(self, caption)
src/graph/graph.py:55
Method__init__
(self, caption='')
src/graph/graph.py:63
Method__init__
(self)
src/graph/graph.py:92
Method__init__
(self, node1, node2)
src/graph/graph.py:133
Method__init__
(self, center_node)
src/graph/graph.py:153
Method__init__
(self, args, is_navigation=True)
src/graph/graph.py:176
Method__init__
(self, llm=None)
src/graph/goalgraphdecomposer.py:4
Method__init__
(self, llm_function)
src/graph/scenegraphcorrector.py:4
Method__init__
(self, llm=None)
src/graph/graphbuilder.py:2
Method__init__
(self, args, envs)
src/agent/unigoal/agent.py:33
Method__lt__
(self, other)
src/graph/graph.py:72
Method__lt__
(self, other)
src/graph/graph.py:104
Methodadd_boundary
(mat, value=1)
src/graph/graph.py:871
Methodapply_operation
(self, operation, G, **kwargs)
src/graph/overlap.py:69
Functionbin_points
Bins points into xy-z bins XYZ_cms is ... x H x W x3 Outputs is ... x map_size x map_size x (len(z_bins)+1)
src/utils/fmm/depth_utils.py:100
Methodbuild_graph
(self, objects, relations)
src/graph/goalgraphdecomposer.py:8
Functionca_short_term_goal
obstacle map (H, W): 1 means occupied, 0 means free start (2, ): planning start top_k_goal (k, 2): fmm generated distance field top k sho
src/utils/fmm/pose_utils.py:116
Functioncartesian_to_polar
(x, y)
src/utils/fmm/pose_utils.py:69
Functioncheck_the_same_floor
(posA, posB)
src/utils/fmm/pose_utils.py:9
Methodcolor_by_instance
(self)
src/graph/utils/slam_classes.py:95
Methodcolor_by_most_common_classes
Color the point cloud of each detection by the most common class
src/graph/utils/slam_classes.py:84
Functioncompress_sem_map
(sem_map)
src/utils/visualization/semantic_prediction.py:54
Methodcompute_similarities
The input feature should be of shape (D, ), a one-row vector This is mostly for backward compatibility
src/graph/utils/slam_classes.py:114
Methodcreate_new_edge
(self, new_node)
src/graph/graph.py:558
Methoddelete_boundary
(mat)
src/graph/graph.py:877
Methodforward
(self, x)
src/utils/model.py:47
Methodforward
(self, obs, pose_obs, maps_last, poses_last, agent_heights)
src/map/bev_mapping.py:59
← previousnext →101–200 of 241, ranked by callers