Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GuangxingHan/QA-FewDet
/ functions
Functions
304 in github.com/GuangxingHan/QA-FewDet
⨍
Functions
304
◇
Types & classes
36
↓ 11 callers
Function
get_layer_param
(model)
QA_FewDet/utils/measures.py:35
↓ 11 callers
Function
get_layer_param
(model)
fewx/utils/measures.py:35
↓ 7 callers
Method
_shared_roi_transform
(self, features, boxes)
QA_FewDet/modeling/fsod/fsod_roi_heads.py:112
↓ 6 callers
Method
_calculate_ap
(self, class_names, precisions, T=None, A=None)
QA_FewDet/evaluation/coco_evaluation.py:250
↓ 6 callers
Method
_calculate_ap
(self, class_names, precisions, T=None, A=None)
fewx/evaluation/coco_evaluation.py:250
↓ 6 callers
Method
_shared_roi_transform
(self, features, boxes)
fewx/modeling/fsod/fsod_roi_heads.py:111
↓ 5 callers
Method
roi_pooling
(self, features, boxes)
QA_FewDet/modeling/fsod/fsod_roi_heads.py:116
↓ 4 callers
Method
init_support_features
(self, evaluation_dataset, evaluation_shot, keepclasses, test_seeds)
fewx/modeling/fsod/fsod_rcnn.py:75
↓ 3 callers
Method
_process_per_class
(self, pos_detector_proposals, query_features, pos_support_box_features, image_size)
QA_FewDet/modeling/fsod/fsod_rcnn.py:197
↓ 3 callers
Method
label_and_sample_proposals
Prepare some proposals to be used to train the ROI heads. It performs box matching between `proposals` and `targets`, and assigns
QA_FewDet/modeling/fsod/fsod_roi_heads.py:124
↓ 3 callers
Method
roi_pooling
(self, features, boxes)
fewx/modeling/fsod/fsod_roi_heads.py:115
↓ 3 callers
Method
ss_edge
(self, support_feature_ls, support_cls_list=[])
QA_FewDet/modeling/fsod/fsod_rcnn.py:240
↓ 2 callers
Function
build_detection_test_loader
Similar to `build_detection_train_loader`. But this function uses the given `dataset_name` argument (instead of the names in cfg), and us
fewx/data/build.py:176
↓ 2 callers
Function
build_detection_train_loader
A data loader is created by the following steps: 1. Use the dataset names in config to query :class:`DatasetCatalog`, and obtain a list of di
fewx/data/build.py:121
↓ 2 callers
Function
build_optimizer
Build an optimizer from config.
fewx/solver/build.py:98
↓ 2 callers
Method
device
(self)
QA_FewDet/modeling/fsod/fsod_rcnn.py:110
↓ 2 callers
Method
device
(self)
fewx/modeling/fsod/fsod_rcnn.py:87
↓ 2 callers
Function
fsod_fast_rcnn_inference
Call `fsod_fast_rcnn_inference_single_image` for all images. Args: boxes (list[Tensor]): A list of Tensors of predicted class-specif
QA_FewDet/modeling/fsod/fsod_fast_rcnn.py:50
↓ 2 callers
Function
fsod_fast_rcnn_inference
Call `fsod_fast_rcnn_inference_single_image` for all images. Args: boxes (list[Tensor]): A list of Tensors of predicted class-specif
fewx/modeling/fsod/fsod_fast_rcnn.py:50
↓ 2 callers
Function
get_cfg
Get a copy of the default config. Returns: a detectron2 CfgNode instance.
fewx/config/config.py:4
↓ 2 callers
Method
inference
Run inference on the given inputs. Args: batched_inputs (list[dict]): same as in :meth:`forward` detected_ins
QA_FewDet/modeling/fsod/fsod_rcnn.py:813
↓ 2 callers
Method
inference
Run inference on the given inputs. Args: batched_inputs (list[dict]): same as in :meth:`forward` detected_ins
fewx/modeling/fsod/fsod_rcnn.py:429
↓ 2 callers
Function
is_leaf
(model)
QA_FewDet/utils/measures.py:25
↓ 2 callers
Function
is_leaf
(model)
fewx/utils/measures.py:25
↓ 2 callers
Method
load_prototype_coco
(self)
QA_FewDet/modeling/fsod/fsod_rcnn.py:737
↓ 2 callers
Method
load_prototype_voc
(self, voc_split_id)
QA_FewDet/modeling/fsod/fsod_rcnn.py:582
↓ 2 callers
Method
losses
Return the losses from a set of RPN predictions and their associated ground-truth. Args: anchors (list[Boxes or RotatedB
QA_FewDet/modeling/fsod/fsod_rpn.py:317
↓ 2 callers
Method
losses
Return the losses from a set of RPN predictions and their associated ground-truth. Args: anchors (list[Boxes or RotatedB
fewx/modeling/fsod/fsod_rpn.py:317
↓ 2 callers
Method
predict_proposals
Decode all the predicted box regression deltas to proposals. Find the top proposals by applying NMS and removing boxes that are too s
QA_FewDet/modeling/fsod/fsod_rpn.py:435
↓ 2 callers
Method
predict_proposals
Decode all the predicted box regression deltas to proposals. Find the top proposals by applying NMS and removing boxes that are too s
fewx/modeling/fsod/fsod_rpn.py:435
↓ 2 callers
Method
preprocess_image
Normalize, pad and batch the input images.
QA_FewDet/modeling/fsod/fsod_rcnn.py:876
↓ 2 callers
Method
preprocess_image
Normalize, pad and batch the input images.
fewx/modeling/fsod/fsod_rcnn.py:489
↓ 1 callers
Method
__init__
NOTE: this interface is experimental. Args: in_features (list[str]): list of names of input features to use
QA_FewDet/modeling/fsod/fsod_rpn.py:154
↓ 1 callers
Method
__init__
Args: box2box_transform (Box2BoxTransform/Box2BoxTransformRotated): box2box transform instance for proposal-to-de
QA_FewDet/modeling/fsod/fsod_fast_rcnn.py:155
↓ 1 callers
Method
__init__
(self)
QA_FewDet/modeling/fsod/gcn_module.py:44
↓ 1 callers
Method
__init__
NOTE: this interface is experimental. Args: in_features (list[str]): list of names of input features to use
fewx/modeling/fsod/fsod_rpn.py:154
↓ 1 callers
Method
__init__
Args: box2box_transform (Box2BoxTransform/Box2BoxTransformRotated): box2box transform instance for proposal-to-de
fewx/modeling/fsod/fsod_fast_rcnn.py:155
↓ 1 callers
Method
_build_res5_block
(self, cfg)
QA_FewDet/modeling/fsod/fsod_roi_heads.py:86
↓ 1 callers
Method
_build_res5_block
(self, cfg)
fewx/modeling/fsod/fsod_roi_heads.py:85
↓ 1 callers
Function
_check_size_scale_factor
(dim)
QA_FewDet/layers/misc.py:35
↓ 1 callers
Function
_check_size_scale_factor
(dim)
fewx/layers/misc.py:35
↓ 1 callers
Function
_create_gradient_clipper
Creates gradient clipping closure to clip by value or by norm, according to the provided config.
QA_FewDet/solver/build.py:24
↓ 1 callers
Function
_create_gradient_clipper
Creates gradient clipping closure to clip by value or by norm, according to the provided config.
fewx/solver/build.py:24
↓ 1 callers
Method
_decode_proposals
Transform anchors into proposals by applying the predicted anchor deltas. Returns: proposals (list[Tensor]): A list of L
QA_FewDet/modeling/fsod/fsod_rpn.py:466
↓ 1 callers
Method
_decode_proposals
Transform anchors into proposals by applying the predicted anchor deltas. Returns: proposals (list[Tensor]): A list of L
fewx/modeling/fsod/fsod_rpn.py:466
↓ 1 callers
Method
_derive_coco_results
Derive the desired score numbers from summarized COCOeval. Args: coco_eval (None or COCOEval): None represents no predic
QA_FewDet/evaluation/coco_evaluation.py:286
↓ 1 callers
Method
_derive_coco_results
Derive the desired score numbers from summarized COCOeval. Args: coco_eval (None or COCOEval): None represents no predic
fewx/evaluation/coco_evaluation.py:286
↓ 1 callers
Method
_eval_box_proposals
Evaluate the box proposals in predictions. Fill self._results with the metrics for "box_proposals" task.
QA_FewDet/evaluation/coco_evaluation.py:211
↓ 1 callers
Method
_eval_box_proposals
Evaluate the box proposals in predictions. Fill self._results with the metrics for "box_proposals" task.
fewx/evaluation/coco_evaluation.py:211
↓ 1 callers
Method
_eval_predictions
Evaluate predictions on the given tasks. Fill self._results with the metrics of the tasks.
QA_FewDet/evaluation/coco_evaluation.py:163
↓ 1 callers
Method
_eval_predictions
Evaluate predictions on the given tasks. Fill self._results with the metrics of the tasks.
fewx/evaluation/coco_evaluation.py:163
↓ 1 callers
Function
_evaluate_box_proposals
Evaluate detection proposal recall metrics. This function is a much faster alternative to the official COCO API recall evaluation code. Howev
QA_FewDet/evaluation/coco_evaluation.py:470
↓ 1 callers
Function
_evaluate_box_proposals
Evaluate detection proposal recall metrics. This function is a much faster alternative to the official COCO API recall evaluation code. Howev
fewx/evaluation/coco_evaluation.py:470
↓ 1 callers
Function
_evaluate_predictions_on_coco
Evaluate the coco results using COCOEval API.
QA_FewDet/evaluation/coco_evaluation.py:582
↓ 1 callers
Function
_evaluate_predictions_on_coco
Evaluate the coco results using COCOEval API.
fewx/evaluation/coco_evaluation.py:582
↓ 1 callers
Function
_generate_optimizer_class_with_gradient_clipping
Dynamically creates a new type that inherits the type of a given instance and overrides the `step` method to add gradient clipping
QA_FewDet/solver/build.py:44
↓ 1 callers
Function
_generate_optimizer_class_with_gradient_clipping
Dynamically creates a new type that inherits the type of a given instance and overrides the `step` method to add gradient clipping
fewx/solver/build.py:44
↓ 1 callers
Function
_get_builtin_metadata_pascal_voc
(dataset_name)
QA_FewDet/data/datasets/builtin_meta_pascal_voc.py:49
↓ 1 callers
Function
_get_builtin_metadata_pascal_voc
(dataset_name)
fewx/data/datasets/builtin_meta_pascal_voc.py:49
↓ 1 callers
Function
_get_pascal_voc_fewshot_instances_meta
()
QA_FewDet/data/datasets/builtin_meta_pascal_voc.py:40
↓ 1 callers
Function
_get_pascal_voc_fewshot_instances_meta
()
datasets/pascal_voc/builtin_meta_pascal_voc.py:40
↓ 1 callers
Function
_get_pascal_voc_fewshot_instances_meta
()
fewx/data/datasets/builtin_meta_pascal_voc.py:40
↓ 1 callers
Method
_log_accuracy
Log the accuracy metrics to EventStorage.
QA_FewDet/modeling/fsod/fsod_fast_rcnn.py:209
↓ 1 callers
Method
_log_accuracy
Log the accuracy metrics to EventStorage.
fewx/modeling/fsod/fsod_fast_rcnn.py:209
↓ 1 callers
Function
_output_size
(dim)
QA_FewDet/layers/misc.py:50
↓ 1 callers
Function
_output_size
(dim)
fewx/layers/misc.py:50
↓ 1 callers
Method
_postprocess
Rescale the output instances to the target size.
QA_FewDet/modeling/fsod/fsod_rcnn.py:894
↓ 1 callers
Method
_postprocess
Rescale the output instances to the target size.
fewx/modeling/fsod/fsod_rcnn.py:507
↓ 1 callers
Method
_predict_boxes
Returns: Tensor: A Tensors of predicted class-specific or class-agnostic boxes for all images in a batch. Element
QA_FewDet/modeling/fsod/fsod_fast_rcnn.py:322
↓ 1 callers
Method
_predict_boxes
Returns: Tensor: A Tensors of predicted class-specific or class-agnostic boxes for all images in a batch. Element
fewx/modeling/fsod/fsod_fast_rcnn.py:322
↓ 1 callers
Method
_sample_child_nodes
(self, center_idx, IoU_matrix, distance_matrix)
QA_FewDet/modeling/fsod/fsod_rcnn.py:167
↓ 1 callers
Method
_subsample_labels
Randomly sample a subset of positive and negative examples, and overwrite the label vector to the ignore value (-1) for all elements
QA_FewDet/modeling/fsod/fsod_rpn.py:242
↓ 1 callers
Method
_subsample_labels
Randomly sample a subset of positive and negative examples, and overwrite the label vector to the ignore value (-1) for all elements
fewx/modeling/fsod/fsod_rpn.py:242
↓ 1 callers
Method
_tasks_from_config
Returns: tuple[str]: tasks that can be evaluated under the given configuration.
QA_FewDet/evaluation/coco_evaluation.py:102
↓ 1 callers
Method
_tasks_from_config
Returns: tuple[str]: tasks that can be evaluated under the given configuration.
fewx/evaluation/coco_evaluation.py:102
↓ 1 callers
Method
build_evaluator
(cls, cfg, dataset_name, output_folder=None)
fsod_train_net.py:81
↓ 1 callers
Method
build_evaluator
(cls, cfg, dataset_name, output_folder=None)
fsod_train_net_fewx.py:81
↓ 1 callers
Function
build_roi_heads
Build ROIHeads defined by `cfg.MODEL.ROI_HEADS.NAME`.
QA_FewDet/modeling/fsod/fsod_roi_heads.py:45
↓ 1 callers
Function
build_roi_heads
Build ROIHeads defined by `cfg.MODEL.ROI_HEADS.NAME`.
fewx/modeling/fsod/fsod_roi_heads.py:44
↓ 1 callers
Function
build_rpn_head
Build an RPN head defined by `cfg.MODEL.RPN.HEAD_NAME`.
QA_FewDet/modeling/fsod/fsod_rpn.py:64
↓ 1 callers
Function
build_rpn_head
Build an RPN head defined by `cfg.MODEL.RPN.HEAD_NAME`.
fewx/modeling/fsod/fsod_rpn.py:64
↓ 1 callers
Method
build_test_loader
Returns: iterable It now calls :func:`detectron2.data.build_detection_test_loader`. Overwrite it if you'd like a
fsod_train_net.py:61
↓ 1 callers
Method
build_test_loader
Returns: iterable It now calls :func:`detectron2.data.build_detection_test_loader`. Overwrite it if you'd like a
fsod_train_net_fewx.py:61
↓ 1 callers
Method
cal_IoU_distance_matrix
(self, proposal_boxes)
QA_FewDet/modeling/fsod/fsod_rcnn.py:162
↓ 1 callers
Function
crop_support
(img, bbox)
datasets/pascal_voc/3_gen_support_pool_any_shot.py:50
↓ 1 callers
Function
crop_support
(img, bbox)
datasets/pascal_voc/2_gen_support_pool.py:50
↓ 1 callers
Function
crop_support
(img, bbox)
datasets/coco/3_gen_support_pool_any_shot_novel_class.py:49
↓ 1 callers
Function
crop_support
(img, bbox)
datasets/coco/4_gen_support_pool_any_shot_full_class.py:49
↓ 1 callers
Function
crop_support
(img, bbox)
datasets/coco/2_gen_support_pool.py:49
↓ 1 callers
Method
eval_with_support
See :meth:`ROIHeads.forward`.
QA_FewDet/modeling/fsod/fsod_roi_heads.py:223
↓ 1 callers
Method
eval_with_support
See :meth:`ROIHeads.forward`.
fewx/modeling/fsod/fsod_roi_heads.py:144
↓ 1 callers
Method
evaluate
(self)
QA_FewDet/evaluation/coco_evaluation.py:134
↓ 1 callers
Method
evaluate
(self)
fewx/evaluation/coco_evaluation.py:134
↓ 1 callers
Function
filter_coco
(coco, cls_split)
datasets/coco/1_split_filter.py:28
↓ 1 callers
Method
forward
Args: pred: Nx4 predicted bounding boxes target: Nx4 target bounding boxes weight: N loss weight for each
QA_FewDet/layers/iou_loss.py:17
↓ 1 callers
Method
forward
Args: pred: Nx4 predicted bounding boxes target: Nx4 target bounding boxes weight: N loss weight for each
fewx/layers/iou_loss.py:17
↓ 1 callers
Method
forward_with_given_boxes
Use the given boxes in `instances` to produce other (non-box) per-ROI outputs. Args: features: same as in `forward()`
QA_FewDet/modeling/fsod/fsod_roi_heads.py:270
↓ 1 callers
Method
forward_with_given_boxes
Use the given boxes in `instances` to produce other (non-box) per-ROI outputs. Args: features: same as in `forward()`
fewx/modeling/fsod/fsod_roi_heads.py:200
↓ 1 callers
Function
fsod_fast_rcnn_inference_single_image
Single-image inference. Return bounding-box detection results by thresholding on scores and applying non-maximum suppression (NMS). Args
QA_FewDet/modeling/fsod/fsod_fast_rcnn.py:85
↓ 1 callers
Function
fsod_fast_rcnn_inference_single_image
Single-image inference. Return bounding-box detection results by thresholding on scores and applying non-maximum suppression (NMS). Args
fewx/modeling/fsod/fsod_fast_rcnn.py:85
next →
1–100 of 304, ranked by callers