Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/bagh2178/UniGoal
/ functions
Functions
241 in github.com/bagh2178/UniGoal
⨍
Functions
241
◇
Types & classes
26
↓ 1 callers
Method
goal_decomposition
(self, goalgraph=None)
src/graph/goalgraphdecomposer.py:33
↓ 1 callers
Function
gobs_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 callers
Method
graph_corr
(self, goal, graph)
src/graph/graph.py:921
↓ 1 callers
Method
graph_corr
(self, nodes_to_check, edges_to_check)
src/graph/scenegraphcorrector.py:51
↓ 1 callers
Method
graph_to_text
(self, graph)
src/graph/goalgraphdecomposer.py:24
↓ 1 callers
Method
init_map_and_pose
(self)
src/map/bev_mapping.py:289
↓ 1 callers
Method
init_map_and_pose_for_env
(self, env_idx=0)
src/map/bev_mapping.py:316
↓ 1 callers
Function
init_vis_image
(goal_name, args)
src/utils/visualization/visualization.py:33
↓ 1 callers
Method
insert_goal
(self, goal=None)
src/graph/graph.py:655
↓ 1 callers
Method
instance_discriminator
(self, planner_inputs, id_lo_whwh_speci)
src/agent/unigoal/agent.py:195
↓ 1 callers
Method
is_edge_reasonable
(self, edge, source_node, target_node)
src/graph/scenegraphcorrector.py:42
↓ 1 callers
Method
is_node_reasonable
(self, node, connected_nodes, connected_edges)
src/graph/scenegraphcorrector.py:30
↓ 1 callers
Function
line_list
(text, line_length=22)
src/utils/visualization/visualization.py:100
↓ 1 callers
Function
main
()
main.py:57
↓ 1 callers
Method
mapping3d
(self)
src/graph/graph.py:454
↓ 1 callers
Function
mask_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 callers
Function
merge_detections_to_objects
( cfg, detection_list: DetectionList, objects: MapObjectList, agg_sim: torch.Tensor )
src/graph/utils/mapping.py:29
↓ 1 callers
Function
merge_obj2_into_obj1
Merge the new object to the old object This operation is done in-place
src/graph/utils/utils.py:432
↓ 1 callers
Method
move_local_map
(self, env_idx=0)
src/map/bev_mapping.py:246
↓ 1 callers
Function
moving_avg
(a, n=2)
src/utils/fmm/fmm_planner.py:53
↓ 1 callers
Method
node_similarity
(self, G1, G2)
src/graph/overlap.py:30
↓ 1 callers
Method
overlap
(self)
src/graph/graph.py:714
↓ 1 callers
Method
parse_text_description
(self, description)
src/graph/graphbuilder.py:31
↓ 1 callers
Function
pcd_denoise_dbscan
(pcd: o3d.geometry.PointCloud, eps=0.02, min_points=10)
src/graph/utils/utils.py:364
↓ 1 callers
Method
predict_remaining_node_positions
(self, common_nodes, positions, scene_graph)
src/graph/overlap.py:123
↓ 1 callers
Method
preprocess_depth
(self, depth, min_d, max_d)
src/agent/unigoal/agent.py:647
↓ 1 callers
Function
r_between
(v_from_, v_to_)
src/utils/fmm/rotation_utils.py:40
↓ 1 callers
Method
reasonableness_correction
(self)
src/graph/graph.py:740
↓ 1 callers
Function
remove_small_frontiers
(frontier_map, min_size=1)
src/utils/map.py:4
↓ 1 callers
Function
resize_gobs
( gobs, image )
src/graph/utils/utils.py:76
↓ 1 callers
Method
run_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 callers
Function
save_video
(image_list, video_path, fps=30, resize_factor=1.0, input_color_space="RGB")
src/utils/visualization/save.py:6
↓ 1 callers
Method
save_visualization
(self, video_path)
src/agent/unigoal/agent.py:841
↓ 1 callers
Method
seed
(self, seed)
src/envs/instanceimagegoal_env.py:126
↓ 1 callers
Method
segment2d
(self)
src/graph/graph.py:422
↓ 1 callers
Method
set_center
(self, center)
src/graph/graph.py:128
↓ 1 callers
Method
set_cfg
(self)
src/graph/graph.py:255
↓ 1 callers
Method
set_full_map
(self, full_map)
src/graph/graph.py:302
↓ 1 callers
Method
set_full_pose
(self, full_pose)
src/graph/graph.py:305
↓ 1 callers
Method
set_goal
(self, goal, auto_improve=False)
src/utils/fmm/fmm_planner_policy.py:62
↓ 1 callers
Method
set_goal_cat_id
(self, idx)
src/envs/instanceimagegoal_env.py:129
↓ 1 callers
Method
set_navigate_steps
(self, navigate_steps)
src/graph/graph.py:267
↓ 1 callers
Method
set_object
(self, object)
src/graph/graph.py:124
↓ 1 callers
Method
set_observations
(self, observations)
src/graph/graph.py:276
↓ 1 callers
Function
setup_cfg
(args)
src/utils/visualization/semantic_prediction.py:87
↓ 1 callers
Function
splat_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 callers
Method
text2value
(self, text)
src/graph/graph.py:933
↓ 1 callers
Function
transform_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 callers
Function
transform_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 callers
Method
update_after_reset
(self)
src/envs/instanceimagegoal_env.py:50
↓ 1 callers
Method
update_edge
(self)
src/graph/graph.py:577
↓ 1 callers
Method
update_group
(self)
src/graph/graph.py:638
↓ 1 callers
Method
update_node
(self)
src/graph/graph.py:516
↓ 1 callers
Method
update_scenegraph
(self)
src/graph/graph.py:682
↓ 1 callers
Method
visualize
(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
Method
add_boundary
(mat, value=1)
src/graph/graph.py:871
Method
apply_operation
(self, operation, G, **kwargs)
src/graph/overlap.py:69
Function
bin_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
Method
build_graph
(self, objects, relations)
src/graph/goalgraphdecomposer.py:8
Function
ca_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
Function
cartesian_to_polar
(x, y)
src/utils/fmm/pose_utils.py:69
Function
check_the_same_floor
(posA, posB)
src/utils/fmm/pose_utils.py:9
Method
color_by_instance
(self)
src/graph/utils/slam_classes.py:95
Method
color_by_most_common_classes
Color the point cloud of each detection by the most common class
src/graph/utils/slam_classes.py:84
Function
compress_sem_map
(sem_map)
src/utils/visualization/semantic_prediction.py:54
Method
compute_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
Method
create_new_edge
(self, new_node)
src/graph/graph.py:558
Method
delete_boundary
(mat)
src/graph/graph.py:877
Method
forward
(self, x)
src/utils/model.py:47
Method
forward
(self, obs, pose_obs, maps_last, poses_last, agent_heights)
src/map/bev_mapping.py:59
← previous
next →
101–200 of 241, ranked by callers