MCPcopy Create free account

hub / github.com/Jingkang50/OpenPSG / functions

Functions484 in github.com/Jingkang50/OpenPSG

↓ 1 callersFunctionparse_args
()
ce7454/grade.py:24
↓ 1 callersFunctionparse_line
Parse information from a line in a requirements text file.
setup.py:72
↓ 1 callersFunctionparse_version_info
(version_str)
openpsg/version.py:7
↓ 1 callersMethodprepare_gtpair
(self, local_container)
openpsg/evaluation/sgg_metrics.py:554
↓ 1 callersMethodprepare_test_pairs
(self, det_result)
openpsg/models/relation_heads/approaches/sampling.py:42
↓ 1 callersMethodprocess_ignore_objects
An API used in inference stage for processing the data when some object classes should be ignored.
openpsg/models/relation_heads/relation_head.py:434
↓ 1 callersFunctionprocess_vg_bbox
For VG_150 dataset. Rescales the bbox coords back to the original. Parameters ---------- og_height : int Original image height
openpsg/utils/vis_tools/preprocess.py:98
↓ 1 callersFunctionreadme
()
setup.py:16
↓ 1 callersMethodregister_container
(self, mode)
openpsg/evaluation/sgg_metrics.py:68
↓ 1 callersMethodregister_container
(self, mode)
openpsg/evaluation/sgg_metrics.py:476
↓ 1 callersMethodregister_container
(self, mode)
openpsg/evaluation/sgg_metrics.py:605
↓ 1 callersMethodrelation_infer
(self, pair_reps, union_reps, proj1,
openpsg/models/relation_heads/gps_head.py:45
↓ 1 callersMethodrelation_simple_test
:param img: :param img_meta: :param gt_bboxes: Usually, under the forward (train/val/test), it should not be None. But
openpsg/models/frameworks/sg_rcnn.py:387
↓ 1 callersMethodrelation_simple_test
:param img: :param img_meta: :param gt_bboxes: Usually, under the forward (train/val/test), it should not be None. Bu
openpsg/models/frameworks/sg_panoptic_fpn.py:554
↓ 1 callersFunctionreporthook
https://github.com/tqdm/tqdm.
openpsg/models/relation_heads/approaches/motif_util.py:254
↓ 1 callersMethodroi_rescale
(self, rois, scale_factor)
openpsg/models/roi_extractors/visual_spatial.py:229
↓ 1 callersFunctionsave_results
(results)
tools/grade.py:19
↓ 1 callersMethodsimple_test_given_bboxes
For SGG~SGCLS mode: Given gt boxes, extract its predicted scores and score dists. Without any post-process.
openpsg/models/frameworks/sg_rcnn.py:369
↓ 1 callersMethodsimple_test_given_bboxes
For SGG~SGCLS mode: Given gt boxes, extract its predicted scores and score dists. Without any post-process.
openpsg/models/frameworks/sg_panoptic_fpn.py:536
↓ 1 callersMethodsingle_roi_forward
(self, feats, rois, masks=Non
openpsg/models/roi_extractors/visual_spatial.py:280
↓ 1 callersMethodsort_rois
:param batch_idx: tensor with what index we're on :param confidence: tensor with confidences between [0,1) :param boxes: tens
openpsg/models/relation_heads/approaches/relation_ranker.py:68
↓ 1 callersMethodsort_rois
(self, det_result)
openpsg/models/relation_heads/approaches/motif.py:344
↓ 1 callersMethodsubmit
( self, data_loader: DataLoader, )
ce7454/evaluator.py:61
↓ 1 callersMethodtrain_epoch
(self)
ce7454/trainer.py:43
↓ 1 callersFunctiontranspose_packed_sequence_inds
Get a TxB indices from sorted lengths. Fetch new_inds, split by new_lens, padding to max(new_lens), and stack. Returns: new_inds (np.
openpsg/models/relation_heads/approaches/motif_util.py:48
↓ 1 callersFunctiontriplet2Result
(triplets, use_mask, eval_pan_rels=True)
openpsg/models/frameworks/psgtr.py:16
↓ 1 callersMethodunion_roi_forward
( self, feats, img_metas, rois, rel_pair_idx, masks=None,
openpsg/models/roi_extractors/visual_spatial.py:328
↓ 1 callersMethodvctree_score_net
(self, num_objs, roi_feat, roi_dist)
openpsg/models/relation_heads/approaches/vctree.py:355
↓ 1 callersFunctionvg_evaluation_single
( mode, groundtruths, predictions, iou_thrs, logger, ind_to_predicates, multiple_p
openpsg/evaluation/sgg_eval.py:58
↓ 1 callersFunctionxcycwh_to_xyxy
Convert [xc yc w h] box format to [x1 y1 x2 y2] format.
openpsg/utils/vis_tools/preprocess.py:44
Method__call__
(self, image)
ce7454/dataset.py:19
Method__call__
(self, results)
openpsg/datasets/pipelines/loading.py:62
Method__call__
(self, results)
openpsg/datasets/pipelines/loading.py:160
Method__call__
(self, results)
openpsg/datasets/pipelines/formatting.py:22
Method__getitem__
(self, index)
ce7454/dataset.py:65
Method__getitem__
(self, i)
openpsg/models/relation_heads/approaches/relation_util.py:90
Method__init__
( self, net: nn.Module, k: int, )
ce7454/evaluator.py:9
Method__init__
( self, stage, root='./data/coco/', num_classes=56, )
ce7454/dataset.py:45
Method__init__
(self, net: nn.Module, train_loader: DataLoader, learning_r
ce7454/trainer.py:15
Method__init__
Args: img_rgb: a numpy array of shape (H, W, C), where H and W correspond to the height and width of the image re
openpsg/utils/vis_tools/detectron_viz.py:46
Method__init__
( self, ann_file, pipeline, classes=None, data_roo
openpsg/datasets/sg.py:17
Method__init__
( self, ann_file, pipeline, classes=None, data_roo
openpsg/datasets/psg.py:20
Method__init__
( self, with_bbox=True, with_label=True, with_mask=True,
openpsg/datasets/pipelines/loading.py:80
Method__init__
( self, num_classes, in_channels, object_classes,
openpsg/models/relation_heads/detr4seg_head.py:45
Method__init__
(self, dim, fpn_dims, context_dim)
openpsg/models/relation_heads/detr4seg_head.py:797
Method__init__
(self, query_dim, hidden_dim, num_heads, d
openpsg/models/relation_heads/detr4seg_head.py:870
Method__init__
(self, **kwargs)
openpsg/models/relation_heads/gps_head.py:21
Method__init__
(self, **kwargs)
openpsg/models/relation_heads/motif_head.py:21
Method__init__
( self, num_classes, in_channels, num_relations, o
openpsg/models/relation_heads/psgtr_head.py:34
Method__init__
(self, dim, fpn_dims, context_dim)
openpsg/models/relation_heads/psgtr_head.py:1218
Method__init__
(self, query_dim, hidden_dim, num_heads, d
openpsg/models/relation_heads/psgtr_head.py:1291
Method__init__
(self, **kwargs)
openpsg/models/relation_heads/imp_head.py:19
Method__init__
(self, num_classes, in_channels, num_relations,
openpsg/models/relation_heads/psgformer_head.py:31
Method__init__
(self, dim, fpn_dims, context_dim)
openpsg/models/relation_heads/psgformer_head.py:1157
Method__init__
(self, query_dim, hidden_dim, num_heads, d
openpsg/models/relation_heads/psgformer_head.py:1230
Method__init__
The public parameters that shared by various relation heads are initialized here.
openpsg/models/relation_heads/relation_head.py:20
Method__init__
(self, **kwargs)
openpsg/models/relation_heads/vctree_head.py:22
Method__init__
(self, idx, is_root=False)
openpsg/models/relation_heads/approaches/vctree_util.py:151
Method__init__
(self, idx, score, label=-1, box=None, is_root=False)
openpsg/models/relation_heads/approaches/vctree_util.py:254
Method__init__
(self, idx, node_score, label, box, is_root=False)
openpsg/models/relation_heads/approaches/vctree_util.py:392
Method__init__
(self, config, obj_classes)
openpsg/models/relation_heads/approaches/dmp.py:26
Method__init__
(self, config, obj_classes, rel_classes)
openpsg/models/relation_heads/approaches/imp.py:18
Method__init__
(self, s_cls_cost=dict(type='ClassificationCost', weight=1.), s_reg_cost=dic
openpsg/models/relation_heads/approaches/matcher.py:15
Method__init__
(self, sub_id_cost=dict(type='ClassificationCost', weight=1.), obj_id_cost=d
openpsg/models/relation_heads/approaches/matcher.py:130
Method__init__
(self, input_dim=1024, hidden_dim=512, nl_layer=1, num_out=1)
openpsg/models/relation_heads/approaches/relation_ranker.py:110
Method__init__
(self, in_dim, out_dim, num_layer, dropout=0.0)
openpsg/models/relation_heads/approaches/treelstm_util.py:20
Method__init__
(self, in_dim, out_dim, dropout=0.0)
openpsg/models/relation_heads/approaches/treelstm_util.py:39
Method__init__
(self, in_dim, out_dim, direction, dropout=0.0)
openpsg/models/relation_heads/approaches/treelstm_util.py:62
Method__init__
(self, feat_dim, h_dim, is_pass_embed=False,
openpsg/models/relation_heads/approaches/treelstm_util.py:95
Method__init__
(self, feat_dim, h_dim, is_pass_embed=False,
openpsg/models/relation_heads/approaches/treelstm_util.py:221
Method__init__
(self, config, obj_classes, rel_classes)
openpsg/models/relation_heads/approaches/vctree.py:101
Method__init__
Arguments:
openpsg/models/relation_heads/approaches/relation_util.py:100
Method__init__
(self)
openpsg/models/relation_heads/approaches/relation_util.py:196
Method__init__
(self, type, pos_iou_thr, require_overlap,
openpsg/models/relation_heads/approaches/sampling.py:22
Method__init__
(self, cfg, statistics, eps=1e-3)
openpsg/models/relation_heads/approaches/motif.py:28
Method__init__
(self, config, obj_classes, rel_classes)
openpsg/models/relation_heads/approaches/motif.py:250
Method__init__
(self, k=64)
openpsg/models/relation_heads/approaches/pointnet.py:54
Method__init__
(self, init_c=32, out_c=1024, global_feat=True,
openpsg/models/relation_heads/approaches/pointnet.py:91
Method__init__
(self, k=2, feature_transform=False)
openpsg/models/relation_heads/approaches/pointnet.py:145
Method__init__
(self, k=2, feature_transform=False)
openpsg/models/relation_heads/approaches/pointnet.py:167
Method__init__
(self, encoder=None, decoder1=None, decoder2=None,
openpsg/models/frameworks/dual_transformer.py:22
Method__init__
( self, backbone, rpn_head, roi_head, train_cfg, test_cfg,
openpsg/models/frameworks/sg_rcnn.py:16
Method__init__
(self, backbone, bbox_head, train_cfg=None,
openpsg/models/frameworks/detr4seg.py:40
Method__init__
( self, backbone, neck=None, rpn_head=None, roi_head=None, tra
openpsg/models/frameworks/sg_panoptic_fpn.py:17
Method__init__
(self, backbone, bbox_head, train_cfg=None,
openpsg/models/frameworks/psgtr.py:75
Method__init__
(self, param, **kwargs)
openpsg/models/roi_heads/scene_graph_roi_head.py:9
Method__init__
( self, bbox_roi_layer, in_channels, featmap_strides,
openpsg/models/roi_extractors/visual_spatial.py:38
Method__init__
(self, eps=1e-6, reduction='mean', loss_weight=1.0)
openpsg/models/losses/seg_losses.py:29
Method__init__
(self, reduction='mean', loss_weight=1.0)
openpsg/models/losses/seg_losses.py:48
Method__init__
(self, reduction='mean', loss_weight=1.0)
openpsg/models/losses/seg_losses.py:63
Method__init__
(self, reduction='mean', loss_weight=1.0)
openpsg/models/losses/seg_losses.py:80
Method__init__
(self, result_dict, nogc_result_dict, nogc_thres_num,
openpsg/evaluation/sgg_metrics.py:25
Method__init__
(self, result_dict)
openpsg/evaluation/sgg_metrics.py:285
Method__init__
(self, *args, **kwargs)
openpsg/evaluation/sgg_metrics.py:353
Method__init__
(self, *args, **kwargs)
openpsg/evaluation/sgg_metrics.py:473
Method__init__
(self, result_dict, nogc_result_dict, nogc_thres_num,
openpsg/evaluation/sgg_metrics.py:591
Method__init__
(self, result_dict)
openpsg/evaluation/sgg_metrics.py:855
Method__iter__
(self)
openpsg/models/relation_heads/approaches/relation_util.py:94
Method__len__
(self)
ce7454/dataset.py:62
Method__len__
(self)
openpsg/models/relation_heads/approaches/relation_util.py:86
← previousnext →201–300 of 484, ranked by callers