MCPcopy Create free account

hub / github.com/CAMMA-public/SurgLatentGraph / functions

Functions239 in github.com/CAMMA-public/SurgLatentGraph

⨍ Functions 239◇ Types & classes 63
↓ 82 callersMethodto
(self, device)
model/predictor_heads/modules/gnn_models.py:369
↓ 67 callersMethodto
(self, *args, **kwargs)
model/lg.py:592
↓ 28 callersMethodupdate
(self, pred_imgs, gt_imgs, gt_boxes)
evaluator/CocoMetricRGD.py:462
↓ 21 callersFunctionbuild_mlp
(dim_list, activation='relu', batch_norm='none', dropout=0, final_nonlinearity=True)
model/predictor_heads/modules/layers.py:211
↓ 9 callersMethodto
(self, *args, **kwargs)
model/modified_detectors/sam_detector.py:293
↓ 7 callersFunctionget_activation
(name)
model/predictor_heads/modules/layers.py:38
↓ 7 callersMethodreset_parameters
(self)
model/predictor_heads/modules/gnn_models.py:383
↓ 6 callersMethodfeature_perturbation
(self, node_feats, edge_feats, img_feats, keep_modality)
model/predictor_heads/ds.py:243
↓ 5 callersMethodforward
(self, graph, node_feats, edge_feats, img_feats, edit_graph: bool = False)
model/predictor_heads/ds.py:482
↓ 5 callersFunctionget_normalization_2d
(channels, normalization)
model/predictor_heads/modules/layers.py:26
↓ 4 callersMethod__init__
(self, node_feat_size, edge_feat_size, hidden_dim_size, num_heads, num_layers, act=None, norm=None
model/predictor_heads/modules/gnn_models.py:245
↓ 4 callersMethod_construct_layout
(self, layout_size, classes, boxes, masks=None)
model/deepcvs.py:80
↓ 4 callersMethod_construct_layout
(self, classes, boxes, masks=None)
model/predictor_heads/reconstruction.py:188
↓ 4 callersMethodforward
(self, graph, node_feats, edge_feats, img_feats)
model/predictor_heads/ds.py:207
↓ 3 callersMethod__init__
(self, mstcn_stages, mstcn_layers, mstcn_f_maps, mstcn_f_dim, out_features, mstcn_causal_conv=True
model/predictor_heads/modules/mstcn.py:10
↓ 3 callersMethodextract_lg
(self, batch_inputs: Tensor, batch_data_samples: SampleList, force_perturb: bool = False, losses:
model/lg.py:304
↓ 3 callersMethodpredict
(self, batch_inputs: Tensor, batch_data_samples: SampleList, rescale: bool = True)
model/lg.py:187
↓ 3 callersMethodtransform
(self, results: dict)
datasets/custom_loading.py:73
↓ 2 callersMethod__init__
(self, size)
model/predictor_heads/modules/layers.py:71
↓ 2 callersMethod__init__
(self, item_index)
model/predictor_heads/modules/utils.py:19
↓ 2 callersMethod_build_edges
(self, results: SampleList, nodes_per_img: List, feats: BaseDataElement = None)
model/predictor_heads/graph.py:115
↓ 2 callersMethod_build_gt_edges
(self, results: SampleList)
model/predictor_heads/graph.py:294
↓ 2 callersMethod_construct_graph
(self, feats: BaseDataElement, edges: BaseDataElement, nodes_per_img: List)
model/predictor_heads/graph.py:254
↓ 2 callersMethod_ds_loss
(self, ds_gt, ds_preds, perturbed_ds_preds, loss_consensus, loss_fn, loss_weight, loss_suffix='')
model/predictor_heads/ds.py:350
↓ 2 callersMethod_ds_predict
(self, final_feats)
model/predictor_heads/ds.py:665
↓ 2 callersMethod_forward
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/deepcvs_temporal.py:162
↓ 2 callersMethod_forward
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/deepcvs.py:114
↓ 2 callersFunction_get_padding
Helper method to compute padding size
model/predictor_heads/modules/layers.py:117
↓ 2 callersFunction_init_conv
(layer, method)
model/predictor_heads/modules/layers.py:53
↓ 2 callersMethod_predict_edge_classes
(self, graph: BaseDataElement, batch_input_shape: tuple)
model/predictor_heads/graph.py:172
↓ 2 callersMethod_select_edges
(self, edges: BaseDataElement, nodes_per_img: List)
model/predictor_heads/graph.py:182
↓ 2 callersMethod_update_graph
(self, graph: BaseDataElement, dgl_g: dgl.DGLGraph)
model/predictor_heads/graph.py:267
↓ 2 callersMethodactvn
(self, x)
model/predictor_heads/modules/decoder.py:131
↓ 2 callersMethodbox_intersection
(self, boxes1, boxes2)
model/predictor_heads/graph.py:460
↓ 2 callersMethodbox_perturbation
(self, boxes: List[Tensor], image_shape: Tuple, clip_size: int)
model/lg.py:557
↓ 2 callersMethodbox_union
(self, boxes1, boxes2)
model/predictor_heads/graph.py:443
↓ 2 callersMethodbuild_st_graph
(self, graphs: BaseDataElement, clip_results: SampleList)
model/sv2lstg.py:239
↓ 2 callersMethoddrop_duplicates
(self, A)
model/predictor_heads/graph.py:422
↓ 2 callersMethodextract_feat
(self, batch_inputs: Tensor, results: SampleList, force_perturb: bool, clip_size: int)
model/lg.py:359
↓ 2 callersMethodextract_feat
(self, batch_inputs: Tensor, batch_data_samples: SampleList, losses: dict = None)
model/sv2lstg.py:582
↓ 2 callersMethodextract_feat
(self, batch_inputs: Tensor)
model/simple_classifier.py:142
↓ 2 callersMethodgenerate_and_decode_prompts
(self, batch_inputs: Tensor, final_filter_key: str, return_logits: bool = True, multimask_output:
model/modified_detectors/sam_detector.py:108
↓ 2 callersMethodget_gram_matrix
(self, x, detach=False)
model/predictor_heads/modules/loss.py:204
↓ 2 callersMethodget_keyframes_per_video
Get all keyframes in one video. Args: idx (int): Index of video. Returns: List[int]: a list of all keyframes
datasets/custom_loading.py:132
↓ 2 callersFunctionget_out_channel
(layer)
model/predictor_heads/modules/SPADE/normalization.py:18
↓ 2 callersMethodget_queries
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/modified_detectors/mask2former_with_queries.py:27
↓ 2 callersMethodloss_and_predict
(self, results: SampleList, feats: BaseDataElement)
model/predictor_heads/graph.py:491
↓ 2 callersMethodmasks_to_polygons
(self, masks: List[Tensor])
model/lg.py:536
↓ 2 callersMethodmatch_edges
Args: pred_boxes_A: List of tensors of length B, where each tensor has shape (N, 4) representing predicted bounding boxes
model/predictor_heads/graph.py:579
↓ 2 callersMethodpredict
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/deepcvs.py:132
↓ 2 callersFunctionprocess_preds
(result)
run_inf_on_vid.py:11
↓ 2 callersMethodreconstruct
(self, batch_inputs: Tensor, results: SampleList, ds_feats: Tensor, ds_bb_feats: List[Tensor])
model/deepcvs.py:146
↓ 2 callersMethodreconstruct
(self, batch_inputs: Tensor, feats: Tensor)
model/simple_classifier.py:86
↓ 2 callersMethodresize_to_sam
(self, imgs: Tensor)
model/modified_detectors/sam_detector.py:100
↓ 1 callersMethod__call__
(self, pred_imgs, gt_imgs, gt_boxes)
evaluator/CocoMetricRGD.py:447
↓ 1 callersMethod__init__
(self, num_classes: int, gnn_cfg: ConfigType, img_feat_key: str, img_feat_size: int, input_viz_fea
model/predictor_heads/ds.py:35
↓ 1 callersMethod__init__
(self, dims, normalization='instance', activation='leakyrelu', spade_blocks=False, source_image_dims=32)
model/predictor_heads/reconstruction.py:320
↓ 1 callersMethod__init__
(self, layout_dim, input_dim, output_dim, normalization='instance', activation='leakyrelu')
model/predictor_heads/modules/decoder.py:34
↓ 1 callersMethod__init__
(self, data_range, size_average, channel)
evaluator/CocoMetricRGD.py:444
↓ 1 callersMethod__init__
(self, saved_graph_dir: str = '', load_keyframes_only: bool = False, skip_keys: List = [], **kwarg
datasets/custom_loading.py:66
↓ 1 callersMethod_box_union
Compute the union of boxes in two lists. Args: boxesA (torch.Tensor): Tensor of shape (N, 4) representing the boxes in l
model/sv2lstg.py:448
↓ 1 callersMethod_build_spatial_edges
(self, boxes: Tensor)
model/sv2lstg.py:518
↓ 1 callersMethod_build_visual_edges
(self, graphs: BaseDataElement)
model/sv2lstg.py:473
↓ 1 callersMethod_compute_iou
(self, roi, rois, area, areas)
model/sv2lstg.py:568
↓ 1 callersMethod_compute_st_sem_feats
(self, boxes: Tensor, class_logits: Tensor, edge_temporal_windows: Tensor, box_shape: Tuple)
model/sv2lstg.py:426
↓ 1 callersMethod_construct_reconstruction_input
(self, images, node_features, layouts, gt_layouts, img_feats)
model/predictor_heads/reconstruction.py:223
↓ 1 callersMethod_create_dgl_graph
(self, graph: BaseDataElement)
model/predictor_heads/modules/gnn.py:60
↓ 1 callersMethod_create_temporal_model
(self)
model/deepcvs_temporal.py:40
↓ 1 callersMethod_create_temporal_model
(self)
model/predictor_heads/ds.py:632
↓ 1 callersMethod_edge_flats_from_adj_mat
(self, presence_logits, nodes_per_img)
model/predictor_heads/graph.py:203
↓ 1 callersMethod_edit_graph
(self, batched_graph, nodes_per_img)
model/predictor_heads/ds.py:678
↓ 1 callersMethod_featurize_st_graph
(self, spat_graph: Tensor, viz_graph: Tensor, node_boxes: Tensor, node_labels: Tensor, graphs: Bas
model/sv2lstg.py:261
↓ 1 callersMethod_init_graph_viz_info
(self)
visualizer/LatentGraphVisualizer.py:47
↓ 1 callersMethod_load_ds
Private function to load downstream annotations. Args: results (dict): Result dict from :obj:``mmengine.BaseDataset``.
datasets/custom_loading.py:31
↓ 1 callersMethod_load_ds
Private function to load downstream annotations. Args: results (dict): Result dict from :obj:``mmengine.BaseDataset``.
datasets/custom_loading.py:50
↓ 1 callersMethod_predict_edge_presence
(self, node_features, nodes_per_img)
model/predictor_heads/graph.py:82
↓ 1 callersMethod_rescale_results
(self, results: SampleList)
model/predictor_heads/reconstruction.py:278
↓ 1 callersMethod_whiteout
(self, images, gt_layouts, layouts)
model/predictor_heads/reconstruction.py:253
↓ 1 callersMethodadd_lg_to_results
(self, results: SampleList, feats: BaseDataElement, graph: BaseDataElement)
model/lg.py:255
↓ 1 callersMethodadd_scene_graph_to_results
(self, results: SampleList, gt_edges: BaseDataElement, graph: BaseDataElement)
model/lg.py:235
↓ 1 callersFunctionapply_sparse_mask
(mat, mask)
model/predictor_heads/modules/utils.py:54
↓ 1 callersMethodbox_area
(self, boxes)
model/predictor_heads/graph.py:435
↓ 1 callersMethodboxes_to_mask
(self, boxes, image_shape)
model/predictor_heads/modules/loss.py:197
↓ 1 callersMethodcalibrate_thresholds
(self, pred_ds: torch.Tensor, gt_ds: torch.Tensor)
evaluator/CocoMetricRGD.py:428
↓ 1 callersMethodclassify_instances
(self, batch_outputs: MaskData)
model/modified_detectors/sam_detector.py:193
↓ 1 callersMethodcompute
(self)
evaluator/CocoMetricRGD.py:466
↓ 1 callersMethodcompute_lg_semantic_feat
(self, lg_list: List, graphs: BaseDataElement)
model/sv2lstg.py:713
↓ 1 callersMethodcompute_semantic_feat
(self, results: SampleList, feats: BaseDataElement, graph: BaseDataElement)
model/lg.py:466
↓ 1 callersMethodcount_parameters
(self, model)
hooks/custom_hooks.py:51
↓ 1 callersMethodcrop_boxes
(self, pred_imgs, gt_imgs, boxes, min_size=12)
model/predictor_heads/modules/loss.py:269
↓ 1 callersMethodcrop_boxes
(self, pred_imgs, gt_imgs, boxes, min_size=12)
evaluator/CocoMetricRGD.py:472
↓ 1 callersMethoddeep_loss
(self, reconstructed_imgs, orig_imgs)
model/predictor_heads/modules/loss.py:215
↓ 1 callersFunctiondense_mask_to_polygon_mask
(dense_instance_mask, N)
model/predictor_heads/modules/utils.py:95
↓ 1 callersMethodedge_classifier_loss
(self, edges, gt_edges, num_training_samples)
model/predictor_heads/graph.py:558
↓ 1 callersMethodedge_presence_loss
(self, presence_logits, edges, gt_edges, num_training_samples)
model/predictor_heads/graph.py:532
↓ 1 callersMethodencode_masks
(self, raw_masks)
visualizer/SAMQueryVisualizer.py:40
↓ 1 callersMethodextract_feat
(self, batch_inputs: Tensor, results: SampleList)
model/deepcvs.py:201
↓ 1 callersMethodextract_feat
(self, batch_inputs: Tensor, return_all: bool = False)
model/modified_detectors/mask2former_with_queries.py:15
↓ 1 callersMethodextract_feat
(self, batch_inputs: Tensor, return_all: bool = False)
model/modified_detectors/def_detr_with_queries.py:13
↓ 1 callersMethodgraphs_to_networkx
(self, data_sample: DetDataSample, img_prefix: str, pred_score_thr: float)
visualizer/LatentGraphVisualizer.py:166
next →1–100 of 239, ranked by callers