MCPcopy Create free account

hub / github.com/MasterBin-IIAU/Unicorn / functions

Functions1,452 in github.com/MasterBin-IIAU/Unicorn

↓ 2 callersFunctionget_dataset
Get a single or set of datasets.
external/lib/test/evaluation/datasets.py:40
↓ 2 callersMethodget_evaluator
(self)
unicorn/exp/base_exp.py:47
↓ 2 callersFunctionget_exp_by_file
(exp_file)
unicorn/exp/build.py:10
↓ 2 callersMethodget_filtered_meter
(self, filter_key="time")
unicorn/utils/metric.py:109
↓ 2 callersFunctionget_jittered_box
Jitter the input box args: box - input bounding box (cx, cy, w, h) (4, ) mode - string 'template' or 'search' indicating template
unicorn/data/data_augment.py:470
↓ 2 callersFunctionget_label_map
target: (4, )
external/lib/test/tracker/unicorn_vos.py:221
↓ 2 callersFunctionget_lmdb_handle
(name)
external/lib/utils/lmdb_utils.py:11
↓ 2 callersMethodget_losses
( self, imgs, x_shifts, y_shifts, expanded_strides, labels,
unicorn/models/yolo_head_det.py:331
↓ 2 callersMethodget_losses
( self, imgs, x_shifts, y_shifts, expanded_strides, labels,
unicorn/models/unicorn_head.py:484
↓ 2 callersMethodget_mask_results
get detection results
external/lib/test/tracker/unicorn_vos.py:123
↓ 2 callersFunctionget_mosaic_coordinate
(mosaic_image, mosaic_index, xc, yc, w, h, input_h, input_w)
unicorn/data/datasets/mosaicdetection.py:17
↓ 2 callersMethodget_mot_dataset
(self)
unicorn/exp/unicorn_track_mask.py:203
↓ 2 callersMethodget_mot_dataset
(self)
unicorn/exp/unicorn_track.py:301
↓ 2 callersMethodget_num_sequences
(self)
unicorn/data/datasets/tracking_net.py:180
↓ 2 callersMethodget_num_sequences
(self)
unicorn/data/datasets/got10k.py:77
↓ 2 callersMethodget_num_sequences
(self)
unicorn/data/datasets/lasot.py:81
↓ 2 callersMethodget_obj_sizes_per_object
Object pixel counts, grouped by object
external/lib/train/dataset/vos_base.py:192
↓ 2 callersMethodget_offsets
(self, img)
external/qdtrack/qdtrack/datasets/pipelines/transforms.py:118
↓ 2 callersMethodget_parameters
Get parameters.
external/lib/test/evaluation/tracker.py:334
↓ 2 callersMethodget_params
(self)
external/qdtrack/qdtrack/datasets/pipelines/transforms.py:234
↓ 2 callersMethodget_paths_and_bboxes
(self, sequence_info)
external/lib/train/dataset/vos_base.py:286
↓ 2 callersFunctionget_plot_draw_styles
()
external/lib/test/analysis/plot_results.py:12
↓ 2 callersMethodget_sot_dataset
(self)
unicorn/exp/unicorn_track_mask.py:188
↓ 2 callersMethodget_sot_dataset
(self, cache_img=False)
unicorn/exp/unicorn_track.py:280
↓ 2 callersMethodget_summary
(accs, names, metrics=('mota', 'num_switches', 'idp', 'idr', 'idf1', 'precision', 'recall'))
unicorn/evaluators/evaluation.py:90
↓ 2 callersMethodget_vid_ids
(self, vidIds=[])
external/qdtrack/qdtrack/datasets/parsers/coco_video_parser.py:51
↓ 2 callersFunctionget_world_size
()
external/lib/utils/misc.py:385
↓ 2 callersMethodinference
(self, img)
tools/demo.py:136
↓ 2 callersFunctionious
Compute cost based on IoU :type atlbrs: list[tlbr] | np.ndarray :type atlbrs: list[tlbr] | np.ndarray :rtype ious np.ndarray
unicorn/tracker/matching.py:53
↓ 2 callersFunctionjpeg4py_loader
Image reading using jpeg4py https://github.com/ajkxyz/jpeg4py
external/lib/train/data/image_loader.py:34
↓ 2 callersMethodkey_img_sampling
(self, img_ids, interval=1)
external/qdtrack/qdtrack/datasets/coco_video_dataset.py:59
↓ 2 callersFunctionl2_loss
L2 loss. Args: pred (torch.Tensor): The prediction. target (torch.Tensor): The learning target of the prediction. Returns:
external/qdtrack/qdtrack/models/losses/l2_loss.py:8
↓ 2 callersMethodload_img_res_mask
(self, seq_name, frame_id, cur_seq_data)
unicorn/data/datasets/mots_mot.py:159
↓ 2 callersMethodload_mask
(self, index)
unicorn/data/datasets/coco_inst.py:227
↓ 2 callersMethodload_mask
(self, index)
unicorn/data/datasets/coco_mots.py:232
↓ 2 callersMethodload_resized_img
(self, index)
unicorn/data/datasets/coco_inst.py:189
↓ 2 callersMethodload_resized_img
(self, index)
unicorn/data/datasets/voc.py:208
↓ 2 callersMethodload_resized_img
(self, index)
unicorn/data/datasets/coco.py:166
↓ 2 callersMethodmark_removed
(self)
unicorn/tracker/basetrack.py:51
↓ 2 callersMethodmask_heads_forward_with_coords
mask_feats: (1, 8, H//8, W//8), mask_feat_stride=8, mask_head_params: (N, 169), instance_locations: (N, 2), instance_fpn_levels: (N,
unicorn/models/condinst/dynamic_mask_head.py:172
↓ 2 callersFunctionmask_merge_parallel
(track_dicts, img_names, bitmask_base, nproc)
external/qdtrack/qdtrack/core/to_bdd100k/utils.py:41
↓ 2 callersMethodmatch
(self, bboxes, labels, track_feats, frame_id, asso_tau=-1, return_index=False)
unicorn/tracker/quasi_dense_embed_tracker.py:137
↓ 2 callersMethodmixup
(self, origin_img, origin_labels, input_dim)
unicorn/data/datasets/mosaicdetection.py:171
↓ 2 callersFunctionms_deform_attn_core_pytorch
(value, value_spatial_shapes, sampling_locations, attention_weights)
unicorn/models/ops/functions/ms_deform_attn_func.py:41
↓ 2 callersFunctionmulti_gpu_test
Test model with multiple gpus. This method tests model with multiple gpus and collects the results under two different modes: gpu and cpu mod
external/qdtrack/qdtrack/apis/test.py:37
↓ 2 callersMethodnext
(self)
unicorn/data/data_prefetcher.py:55
↓ 2 callersMethodnext_id
()
unicorn/tracker/basetrack.py:35
↓ 2 callersFunctionnms
Single class NMS implemented in Numpy.
unicorn/utils/demo_utils.py:17
↓ 2 callersMethodobject_ids
All objects in the sequence
external/lib/train/dataset/youtubevos.py:46
↓ 2 callersFunctionparse_gts
(gts, is_mot15)
external/qdtrack/tools/convert_datasets/mot2coco.py:61
↓ 2 callersMethodpre_pipeline
Prepare results dict for pipeline.
external/qdtrack/qdtrack/datasets/coco_video_dataset.py:89
↓ 2 callersMethodpreload
(self)
unicorn/data/data_prefetcher.py:43
↓ 2 callersMethodpreload
(self)
unicorn/data/data_prefetcher.py:74
↓ 2 callersMethodpreload
(self)
unicorn/data/data_prefetcher.py:108
↓ 2 callersMethodpreload
(self)
unicorn/data/data_prefetcher.py:143
↓ 2 callersMethodprepare_results
(self, img_info)
external/qdtrack/qdtrack/datasets/coco_video_dataset.py:113
↓ 2 callersFunctionpreprocessResult
Preprocesses data for utils.CLEAR_MOT_M. Returns a subset of the predictions.
external/qdtrack/qdtrack/core/evaluation/mot_pcan.py:38
↓ 2 callersFunctionprint_error
(target, result)
external/davis2017-evaluation/pytest/test_evaluation.py:45
↓ 2 callersMethodprocess
(self, img_arr: np.ndarray, input_size: tuple)
external/lib/test/tracker/unicorn_sot.py:114
↓ 2 callersMethodprocess
(self, img_arr: np.ndarray, input_size: tuple)
external/lib/test/tracker/unicorn_vos.py:207
↓ 2 callersMethodproject
Project state distribution to measurement space. Parameters ---------- mean : ndarray The state's mean vector (8
unicorn/tracker/kalman_filter.py:126
↓ 2 callersFunctionrandom_color
Random a color according to the input seed.
external/qdtrack/qdtrack/core/utils/visualization.py:14
↓ 2 callersFunctionrandom_perspective
( img, targets=(), degrees=10, translate=0.1, scale=0.1, shear=10, perspective=0.0
unicorn/data/data_augment.py:52
↓ 2 callersMethodre_activate
(self, new_track, frame_id, new_id=False)
unicorn/tracker/byte_tracker.py:61
↓ 2 callersMethodread_masks
(self, sequence, masks_id)
external/davis2017-evaluation/davis2017/results.py:22
↓ 2 callersMethodreset
(self)
external/lib/train/admin/stats.py:25
↓ 2 callersMethodreset_accumulator
(self)
unicorn/evaluators/evaluation.py:25
↓ 2 callersMethodroll
(self)
external/lib/train/data/transforms.py:262
↓ 2 callersMethodrun_sequence
Run tracker on sequence. args: seq: Sequence to run the tracker on. visualization: Set visualization flag (None means
external/lib/test/evaluation/tracker.py:70
↓ 2 callersFunctionsample_target
Extracts a square crop centered at target_bb box, of area search_area_factor^2 times target_bb area args: im - cv image target_b
external/lib/train/data/processing_utils.py:12
↓ 2 callersFunctionsave_mask
(mask, img_path)
external/davis2017-evaluation/davis2017/utils.py:127
↓ 2 callersFunctionsave_score
(file, data)
external/lib/test/evaluation/running.py:37
↓ 2 callersFunctionsave_time
(file, data)
external/lib/test/evaluation/running.py:33
↓ 2 callersFunctionseg2bmap
From a segmentation, compute a binary boundary map with 1 pixel wide boundaries. The boundary pixels are offset by 1/2 pixel towards the
external/PyDavis16EvalToolbox/metrics/f_boundary.py:76
↓ 2 callersFunctionsub_stracks
(tlista, tlistb)
unicorn/tracker/byte_tracker.py:313
↓ 2 callersMethodtlbr_to_tlwh
(tlbr)
unicorn/tracker/byte_tracker.py:131
↓ 2 callersMethodtlwh_to_tlbr
(tlwh)
unicorn/tracker/byte_tracker.py:138
↓ 2 callersMethodto_xyah
(self)
unicorn/tracker/byte_tracker.py:126
↓ 2 callersMethodupdate
(self, value, n=1)
external/lib/utils/misc.py:39
↓ 2 callersMethodupdate_lr
(self, iters)
unicorn/utils/lr_scheduler.py:33
↓ 2 callersMethodvisual
(self, output, img_info, cls_conf=0.35)
tools/demo.py:190
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
unicorn/models/backbone/swin_transformer.py:39
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
unicorn/models/transformer_encoder.py:70
↓ 2 callersFunctionwrite_results_mots
(filename, results)
unicorn/evaluators/mot_evaluator.py:37
↓ 1 callersFunctionBuildSortedDetectionList
Helper function to Accumulate() Considers the evaluation results applicable to a particular category, area range, and max_detections parameter setting
unicorn/layers/csrc/cocoeval/cocoeval.cpp:221
↓ 1 callersFunctionComputePrecisionRecallCurve
Helper function to Accumulate() Compute a precision recall curve given a sorted list of detected instances encoded in evaluations, evaluation_indices,
unicorn/layers/csrc/cocoeval/cocoeval.cpp:282
↓ 1 callersMethodInstanceAnnotation
unicorn/layers/csrc/cocoeval/cocoeval.h:16
↓ 1 callersFunctionMatchDetectionsToGroundTruth
For each IOU threshold, greedily match each detected instance to a ground truth instance (if possible) and store the results
unicorn/layers/csrc/cocoeval/cocoeval.cpp:59
↓ 1 callersFunctionSortInstancesByDetectionScore
Sort detections from highest score to lowest, such that detection_instances[detection_sorted_indices[t]] >= detection_instances[detection_sorted_indic
unicorn/layers/csrc/cocoeval/cocoeval.cpp:16
↓ 1 callersFunctionSortInstancesByIgnore
Partition the ground truth objects based on whether or not to ignore them based on area
unicorn/layers/csrc/cocoeval/cocoeval.cpp:32
↓ 1 callersMethod__add__
(self, other)
external/lib/utils/tensor.py:58
↓ 1 callersMethod__call__
(self, results)
external/qdtrack/qdtrack/datasets/pipelines/loading.py:42
↓ 1 callersMethod__call__
(self, results)
external/qdtrack/qdtrack/datasets/pipelines/formatting.py:81
↓ 1 callersMethod__init__
(self, with_ins_id=False, *args, **kwargs)
external/qdtrack/qdtrack/datasets/pipelines/loading.py:22
↓ 1 callersMethod__init__
(self, normalize=False)
external/lib/test/tracker/unicorn_sot.py:112
↓ 1 callersMethod__init__
(self, normalize=False)
external/lib/test/tracker/unicorn_vos.py:205
↓ 1 callersMethod__init__
(self, data=None, filename=None)
external/lib/train/dataset/vos_base.py:14
↓ 1 callersMethod__init__
args: transform - The set of transformations to be applied on the images. Used only if template_transform or
external/lib/train/data/processing.py:18
↓ 1 callersMethod__init__
Args: size (int): the total number of data of the underlying dataset to sample from shuffle (bool): whether to shuffl
unicorn/data/samplers.py:41
↓ 1 callersMethod__init__
Args: dataset(Dataset) : Pytorch dataset object. img_size (tuple): mosaic (bool): enable mosaic augmenta
unicorn/data/datasets/mosaicdetection_uni.py:24
← previousnext →201–300 of 1,452, ranked by callers