MCPcopy Create free account

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

Functions239 in github.com/CAMMA-public/SurgLatentGraph

Methodforward
(self, X)
model/predictor_heads/modules/SPADE/architectures.py:38
Methodforward
(self, x, segmap)
model/predictor_heads/modules/SPADE/normalization.py:96
Methodforward
Forward function
model/roi_extractors/sg_single_level_roi_extractor.py:52
Methodforward
(self, x: List[Tensor], batch_data_samples: SampleList, return_queries: bool =
model/modified_detectors/mask2former_with_queries.py:42
Functionget_nonspade_norm_layer
(opt, norm_type='instance')
model/predictor_heads/modules/SPADE/normalization.py:16
Methodget_queries
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/modified_detectors/def_detr_with_queries.py:25
Functionget_sparse_mask_inds
(mat, inds, N)
model/predictor_heads/modules/utils.py:76
Methodinit_func
(m)
model/predictor_heads/modules/SPADE/base_network.py:29
Methodinit_weights
(self, init_type='normal', gain=0.02)
model/predictor_heads/modules/SPADE/base_network.py:28
Methodinstance_postprocess
Instance segmengation postprocess. Args: mask_cls (Tensor): Classfication outputs of shape (num_queries, cls_out_c
model/modified_detectors/mask2former_with_queries.py:119
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/lg.py:149
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/deepcvs_temporal.py:66
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/deepcvs.py:163
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/sv2lstg.py:101
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/simple_classifier.py:105
Methodloss
(self, graph: BaseDataElement, feats: BaseDataElement, batch_data_samples: SampleList)
model/predictor_heads/ds.py:325
Methodloss
(self, graph: BaseDataElement, feats: BaseDataElement, batch_data_samples: SampleList)
model/predictor_heads/ds.py:561
Methodloss
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/modified_detectors/sam_detector.py:234
Methodmodify_commandline_options
(parser, is_train)
model/predictor_heads/modules/SPADE/base_network.py:15
Methodnum_total_keyframes
Get the number of all the keyframes in this video dataset.
datasets/custom_loading.py:200
Methodparse_data_info
(self, raw_data_info: dict)
datasets/custom_loading.py:118
Methodpredict
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/sv2lstg.py:121
Methodpredict
(self, batch_inputs: Tensor, batch_data_samples: SampleList)
model/simple_classifier.py:63
Methodpredict
(self, results: SampleList, feats: BaseDataElement)
model/predictor_heads/graph.py:227
Methodpredict
(self, results: Tuple, feats: BaseDataElement, graph: BaseDataElement, imgs: Tensor)
model/predictor_heads/reconstruction.py:99
Methodpredict
(self, batch_inputs: Tensor, batch_data_samples: SampleList, final_filter_key: str = 'iou_preds')
model/modified_detectors/sam_detector.py:237
Methodprint_network
(self)
model/predictor_heads/modules/SPADE/base_network.py:18
Methodprocess
(self, data_batch: Dict, data_samples: Sequence[dict])
evaluator/CocoMetricRGD.py:62
Methodreset_parameters
(self)
model/predictor_heads/modules/gnn_models.py:395
Methodreset_parameters
(self)
model/predictor_heads/modules/gnn_models.py:425
Methodrun
Launch validation.
runner/custom_loops.py:6
Methodrun
Launch test.
runner/custom_loops.py:23
Methodsampling_frames
Sampling frames. Args: video_length (int): The length of the video. key_frame_id (int): The key frame id. Re
datasets/custom_loading.py:305
Methodtrain
(self, mode=True)
model/predictor_heads/modules/loss.py:261
Methodtrain
(self, mode=True)
model/predictor_heads/modules/norm.py:65
Methodtrain
(self, mode)
model/predictor_heads/modules/gnn_models.py:376
Methodtransform
(self, results: dict)
datasets/custom_loading.py:43
Methodtransform
(self, results: dict)
datasets/custom_loading.py:59
Methodtransform
(self, video_infos: dict)
datasets/custom_loading.py:358
← previous201–239 of 239, ranked by callers