MCPcopy Create free account

hub / github.com/ChenhongyiYang/QueryDet-PyTorch / functions

Functions164 in github.com/ChenhongyiYang/QueryDet-PyTorch

↓ 9 callersFunctionpermute_to_N_HWA_K
(tensor, K)
models/querydet/detector.py:45
↓ 8 callersFunctionget_save_path
(img_path, index)
visdrone/data_prepare.py:15
↓ 8 callersMethodresume_or_load
If `resume==True` and `cfg.OUTPUT_DIR` contains the last checkpoint (defined by a `last_checkpoint` file), resume from the file. Resu
train_tools/visdrone_train.py:109
↓ 6 callersFunctionpermute_to_N_HWA_K
Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
models/retinanet/retinanet.py:32
↓ 4 callersFunctionadd_custom_config
(cfg)
configs/custom_config.py:5
↓ 3 callersMethod__init__
(self, in_channels, conv_channels, num_convs, pred_channels, pred_prior=None)
models/querydet/det_head.py:92
↓ 3 callersMethod_make_spconv
(self, weights, biases)
models/querydet/qinfer.py:101
↓ 3 callersMethod_run_spconvs
(self, x, filters)
models/querydet/qinfer.py:129
↓ 3 callersMethoddevice
(self)
models/retinanet/retinanet.py:137
↓ 3 callersFunctionrun_function
(start, end, functions)
utils/gradient_checkpoint.py:212
↓ 3 callersMethodtest
(cls, cfg, model, evaluators=None)
train_tools/visdrone_infer.py:71
↓ 2 callersMethod_bn_convert
(self)
models/querydet/det_head.py:223
↓ 2 callersMethod_bn_convert
(self)
models/querydet/det_head.py:322
↓ 2 callersFunction_soft_nms
Soft non-max suppression algorithm. Implementation of [Soft-NMS -- Improving Object Detection With One Line of Codec] (https://arxiv.org
utils/soft_nms.py:203
↓ 2 callersFunctionbbox_overlaps
Calculate the ious between each bbox of bboxes1 and bboxes2. Args: bboxes1(ndarray): shape (n, 4) bboxes2(ndarray): shape (k,
visdrone_eval/viseval/bbox_overlaps.py:6
↓ 2 callersFunctionbuild_test_loader
(cfg)
visdrone/dataloader.py:127
↓ 2 callersFunctionbuild_train_loader
(cfg)
visdrone/dataloader.py:79
↓ 2 callersMethodget_params
(self)
models/querydet/det_head.py:124
↓ 2 callersMethodinference
Arguments: box_cls, box_delta: Same as the output of :meth:`RetinaNetHead.forward` anchors (list[Boxes]): A list of #
models/retinanet/retinanet.py:330
↓ 2 callersFunctionmake_json
(images, annotations, new_label_json)
visdrone/data_prepare.py:87
↓ 2 callersFunctionopen_label_file
(path, dtype=np.float32)
visdrone_eval/evaluate.py:16
↓ 2 callersFunctionparse_require_file
(fpath)
visdrone_eval/setup.py:58
↓ 2 callersMethodpreprocess_image
(self, batched_inputs)
models/querydet/detector.py:492
↓ 2 callersFunctionsetup
Create configs and perform basic setups.
train_tools/coco_infer.py:125
↓ 1 callersMethod__init__
(self, cfg)
models/retinanet/retinanet.py:76
↓ 1 callersMethod_eval_forward
(self, inputs)
utils/merged_sync_bn.py:62
↓ 1 callersMethod_forward_eval
(self, features, lvl_start)
models/querydet/det_head.py:206
↓ 1 callersMethod_forward_eval
(self, features)
models/querydet/det_head.py:312
↓ 1 callersMethod_forward_train
(self, features, lvl_start)
models/querydet/det_head.py:186
↓ 1 callersMethod_forward_train
(self, features)
models/querydet/det_head.py:304
↓ 1 callersMethod_giou_loss
(self, pred_deltas, anchors, gt_boxes)
models/querydet/detector.py:267
↓ 1 callersMethod_grid_anchors
(self, grid_sizes)
utils/anchor_gen.py:13
↓ 1 callersFunction_instances_to_coco_json
Dump an "Instances" object to a COCO-format json that's used for evaluation. Args: instances (Instances): img_id (int): the
utils/coco_eval_fpn.py:4
↓ 1 callersMethod_iou
(self, boxes, box)
utils/loop_matcher.py:24
↓ 1 callersMethod_make_conv
(self, weights, biases)
models/querydet/qinfer.py:115
↓ 1 callersMethod_make_sparse_tensor
(self, query_logits, last_ys, last_xs, anchors, feature_value)
models/querydet/qinfer.py:40
↓ 1 callersMethod_run_convs
(self, x, filters)
models/querydet/qinfer.py:133
↓ 1 callersFunctionbatched_soft_nms
Performs soft non-maximum suppression in a batched fashion. Each index value correspond to a category, and NMS will not be applied betwe
utils/soft_nms.py:87
↓ 1 callersMethodbuild_evaluator
(cls, cfg, dataset_name, output_folder=None)
train_tools/visdrone_infer.py:53
↓ 1 callersMethodbuild_evaluator
(cls, cfg, dataset_name, output_folder=None)
train_tools/visdrone_train.py:136
↓ 1 callersMethodbuild_test_loader
(cls, cfg, dataset_name)
train_tools/visdrone_infer.py:67
↓ 1 callersMethodbuild_test_loader
(cls, cfg, dataset_name)
train_tools/visdrone_train.py:150
↓ 1 callersMethodbuild_train_loader
(cls, cfg)
train_tools/visdrone_infer.py:63
↓ 1 callersMethodbuild_train_loader
(cls, cfg)
train_tools/visdrone_train.py:146
↓ 1 callersFunctionbuild_transform_gen
(cfg, is_train)
visdrone/mapper.py:87
↓ 1 callersFunctioncalc_accuracy
:param num_imgs: int :param all_gt: list of np.array[m, 8], [:, 4] == 1 indicates ignored regions, which should be dro
visdrone_eval/viseval/calc_accuracy.py:77
↓ 1 callersFunctioncheck_backward_validity
(inputs: Iterable[Any])
utils/gradient_checkpoint.py:23
↓ 1 callersFunctioncheckpoint
r"""Checkpoint a model or part of the model Checkpointing works by trading compute for memory. Rather than storing all intermediate activatio
utils/gradient_checkpoint.py:106
↓ 1 callersFunctioncopy_image
(img_root, img_path, new_img_root)
visdrone/data_prepare.py:39
↓ 1 callersFunctioncreate_int_img
(img)
visdrone_eval/viseval/drop_objects_in_igr.py:4
↓ 1 callersFunctioncrop_and_save_image
(img_root, img_path, new_img_root)
visdrone/data_prepare.py:19
↓ 1 callersMethoddecode_dets
(self, cls_results, reg_results, anchors)
models/querydet/detector.py:498
↓ 1 callersFunctiondefault_argument_parser
Create a parser with some common arguments used by detectron2 users. Args: epilog (str): epilog passed to ArgumentParser describing
train_tools/visdrone_infer.py:92
↓ 1 callersFunctiondefault_argument_parser
Create a parser with some common arguments used by detectron2 users. Args: epilog (str): epilog passed to ArgumentParser describing
train_tools/visdrone_train.py:166
↓ 1 callersFunctiondefault_argument_parser
Create a parser with some common arguments used by detectron2 users. Args: epilog (str): epilog passed to ArgumentParser describing
train_tools/coco_infer.py:68
↓ 1 callersFunctiondefault_argument_parser
Create a parser with some common arguments used by detectron2 users. Args: epilog (str): epilog passed to ArgumentParser describing
train_tools/coco_train.py:118
↓ 1 callersMethoddet_loss
(self, gt_classes, gt_anchors_targets, pred_logits, pred_deltas, all_anchors)
models/querydet/detector.py:274
↓ 1 callersMethoddet_loss
(self, gt_classes, gt_anchors_deltas, pred_logits, pred_deltas, alphas, gammas, cls_weights, reg_weights)
models/retinanet/retinanet.py:273
↓ 1 callersFunctiondetach_variable
(inputs: Tuple[Any, ...])
utils/gradient_checkpoint.py:6
↓ 1 callersFunctiondrop_objects_in_igr
(gt, det, img_height, img_width)
visdrone_eval/viseval/drop_objects_in_igr.py:10
↓ 1 callersFunctioneval_det
:param all_gt: list of np.array[m, 8] :param all_det: list of np.array[m, 6], truncation and occlusion not necessary :param allheight:
visdrone_eval/viseval/eval_det.py:5
↓ 1 callersFunctioneval_res
:param gt0: np.array[ng, 5], ground truth results [x, y, w, h, ignore] :param dt0: np.array[nd, 5], detection results [x, y, w, h, score]
visdrone_eval/viseval/calc_accuracy.py:5
↓ 1 callersFunctiongen_packages_items
()
visdrone_eval/setup.py:66
↓ 1 callersFunctionget_anchor_center_min_dis
Args: box_centers: [N, 2] anchor_centers: [M, 2] Returns:
utils/utils.py:11
↓ 1 callersFunctionget_box_scales
(boxes: Boxes)
utils/utils.py:8
↓ 1 callersMethodget_det_gt
(self, anchors, targets)
models/querydet/detector.py:354
↓ 1 callersMethodget_det_gt
(self, anchors, targets)
models/retinanet/retinanet.py:226
↓ 1 callersFunctionget_device_states
(*args)
utils/gradient_checkpoint.py:35
↓ 1 callersFunctionget_new_label
(label, img_path, cy, cx, id, img_id_base)
visdrone/data_prepare.py:46
↓ 1 callersMethodget_query_gt
(self, small_anchor_centers, targets)
models/querydet/detector.py:409
↓ 1 callersFunctionget_test_data_dicts
(json_file, img_root)
visdrone/dataloader.py:63
↓ 1 callersFunctionget_train_data_dicts
(json_file, img_root, filter_empty=False)
visdrone/dataloader.py:31
↓ 1 callersMethodinference
(self, retina_box_cls, retina_box_delta, retina_anchors, small_det_logits
models/querydet/detector.py:431
↓ 1 callersMethodinference_single_image
(self, retina_box_cls, retina_box_delta, retina_anchors,
models/querydet/detector.py:465
↓ 1 callersMethodinference_single_image
Single-image inference. Return bounding-box detection results by thresholding on scores and applying non-maximum suppression (NMS).
models/retinanet/retinanet.py:357
↓ 1 callersFunctionlabel_to_coco
(label, id, img_id)
visdrone/data_prepare.py:81
↓ 1 callersFunctionmain
()
visdrone_eval/evaluate.py:24
↓ 1 callersFunctionmake_new_test_set
(img_root, label_root, new_img_root, new_label_json)
visdrone/data_prepare.py:132
↓ 1 callersFunctionmake_new_train_set
(img_root, label_root, new_img_root, new_label_json)
visdrone/data_prepare.py:107
↓ 1 callersFunctionparse_line
Parse information from a line in a requirements text file
visdrone_eval/setup.py:25
↓ 1 callersFunctionparse_requirements
Parse the package dependencies listed in a requirements file but strips specific versioning information. Args: fname (str):
visdrone_eval/setup.py:5
↓ 1 callersFunctionpermute_all_to_NHWA_K_not_concat
(box_cls, box_delta, num_classes=80)
models/querydet/detector.py:62
↓ 1 callersFunctionpermute_all_to_NHWA_K_not_concat
(box_cls, box_delta, num_classes=80)
models/retinanet/retinanet.py:65
↓ 1 callersMethodpreprocess_image
Normalize, pad and batch the input images.
models/retinanet/retinanet.py:422
↓ 1 callersMethodquery_loss
(self, gt_small_obj, pred_small_obj, gammas, weights)
models/querydet/detector.py:347
↓ 1 callersFunctionread_label_txt
(txt_file)
visdrone/utils.py:7
↓ 1 callersMethodreset
(self)
utils/time_evaluator.py:19
↓ 1 callersMethodreset
(self)
utils/json_evaluator.py:30
↓ 1 callersMethodrun_qinfer
(self, model_params, features_key, features_value, anchors_value)
models/querydet/qinfer.py:136
↓ 1 callersFunctionset_device_states
(devices, states)
utils/gradient_checkpoint.py:49
↓ 1 callersMethodset_low_quality_matches_
(self, match_labels, matched_vals, matches, gt_ious)
utils/loop_matcher.py:70
↓ 1 callersFunctionsetup
Create configs and perform basic setups.
train_tools/visdrone_infer.py:149
↓ 1 callersFunctionsetup
Create configs and perform basic setups.
train_tools/visdrone_train.py:223
↓ 1 callersFunctionsetup
Create configs and perform basic setups.
train_tools/coco_train.py:175
↓ 1 callersFunctionsoft_nms
Performs soft non-maximum suppression algorithm on axis aligned boxes Args: boxes (Tensor[N, 5]): boxes where NMS will be
utils/soft_nms.py:6
↓ 1 callersFunctionsoft_nms_rotated
Performs soft non-maximum suppression algorithm on rotated boxes Args: boxes (Tensor[N, 5]): boxes where NMS will be perf
utils/soft_nms.py:47
↓ 1 callersMethodtest
(cls, cfg, model, evaluators=None)
train_tools/visdrone_train.py:154
↓ 1 callersMethodtest
(self, batched_inputs)
models/querydet/detector.py:208
↓ 1 callersMethodtest_forward
(self, images)
models/querydet/detector.py:221
↓ 1 callersMethodto_txt
(self)
visdrone/json_to_txt.py:25
next →1–100 of 164, ranked by callers